Nick Joseph, Head of Pretraining at Anthropic, shares insights on how AI models are trained at scale, the infrastructure challenges involved, and how the pre-training team has evolved over time.
Background and Journey to Anthropic
Career Path
- Started at Vicarious, originally an AGI lab that shifted to robotics products
- Worked on training computer vision models for robotics - first job in ML
- Learned ML infrastructure fundamentals at Vicarious
- Joined OpenAI on one of the safety teams, worked on code models
- Saw GPT-3 fine-tuned to write code and realized AI could self-improve
- Left OpenAI with the safety team leads to join Anthropic at founding
Early Interest in AI Safety
- Internship at GiveWell (nonprofit evaluating charities) introduced AI safety concepts
- Initially skeptical - AI safety discussions were mostly theoretical at the time
- Models weren’t good enough to pose real dangers yet
- Decided to work on AI: either safety would matter, or he’d build useful AI tools
What is Pre-training?
The Core Concept
- Scale is a key ingredient to making AI models better
- Need an objective with massive amounts of data
- The internet is the biggest single source of data humanity has created
- No manual labels needed - predict the next word from existing text
How It Works
- Take text and predict the next word
- “The cat” -> predict what comes next
- Every word becomes a new training example
- Dense signal from massive data
Scaling Laws
- As you add more compute, data, and parameters, models get predictably better
- Lower loss = better prediction of next word
- Original scaling laws paper showed 11 orders of magnitude of consistent trends
- This creates a positive feedback loop: train model -> make useful product -> get money -> buy more compute -> train better model
Early Days at Anthropic
Starting Small
- Team of about 10-12 people initially
- Much less funding than competitors like OpenAI
- GPT-3 cost about $5 million to train - significant but achievable
- Used cloud provider for compute
Efficiency as Competitive Advantage
- “We have way less funding than everyone else, we have to be really efficient”
- Most people weren’t very efficient with compute at the time
- Focused on getting the most out of available hardware
Infrastructure Challenges
- Had to understand literal physical layout of chips
- Ran clustering algorithms to identify which rooms chips were in
- Network latency between different physical locations caused issues
- Constantly pushing limits of hardware
Distributed Training Approaches
Key Techniques
- Data Parallelism: Split data across multiple chips
- Pipelining: Chain operations across chips
- Op-sharding: Distribute operations across hardware
Building Custom Infrastructure
- No great open-source packages existed at the time
- Wrote their own all-reduce implementation
- Needed to modify everything as they scaled beyond what Facebook had done
- Couldn’t rely on external packages they’d constantly need to modify
Why Build In-House?
- Going to bigger scale than existing solutions supported
- Need full control to optimize at every level
- Can make architecture changes that simplify efficiency problems
Scaling Laws and Conviction
The Physicist’s Perspective
- Original scaling laws paper showed remarkably consistent trends
- Debate about whether trends would continue seemed “nonsensical”
- 11 orders of magnitude of data, arguing about one more point seemed unlikely to fail
- “What are these people doing? They’re all missing the big picture”
Different Lab Cultures
- FAIR (Facebook AI Research) had PhD-style independent research culture
- People fighting for their own compute, pursuing individual ideas
- Training large language models requires collaboration on infrastructure
- Not publishable work - “I got 5% more efficiency” isn’t a paper
- Anthropic’s culture focused on collaborative scaling
Team Evolution Over Time
Early Days
- First 3-6 months: tried to read every PR in the codebase
- Knew all the pieces, could make cross-cutting changes easily
- Could spot efficiency improvements by changing architecture
Growing Specialization
- As team grew, people became deep experts on individual components
- Some focus on exactly how attention should work
- Others dial in parallelism strategy
- Benefits: can go really deep on specific problems
The Trade-off
- Specialists may miss cross-component optimizations
- Manager’s role: ensure bigger picture makes sense
- Need people who understand the whole system
- Avoid single points of failure
Balancing Generalists and Specialists
- People have genuine preferences for generalist vs specialist roles
- Early hires tend to be generalist-shaped (startup mentality)
- Risk: everyone doing everything, no one deeply understanding anything
- Too many specialists: manager has to connect everything
- Goal: balance of both types
What Hasn’t Changed
The Core Metric
- Still pushing down the same loss function from day one
- “Loss go down” remains the one OKR that matters
- Could plot progress on same metric from first model to today
- Answer is always “as low as possible”
The Feedback Loop
- Train model -> deploy -> get revenue -> buy more compute -> repeat
- This cycle has run continuously for ~5 years
- Scaling laws give conviction that progress will continue
Key Takeaways
- Scale matters: More compute, data, and parameters predictably improve models
- Efficiency is competitive advantage: Especially when resource-constrained
- Build what you need to control: Custom infrastructure enables optimization
- Balance team composition: Mix of generalists and specialists
- Stay focused on core metric: Loss reduction drives everything
Nick Joseph,Anthropic预训练负责人,分享了关于大规模AI模型训练、基础设施挑战以及预训练团队演变的见解。
背景与加入Anthropic的历程
职业路径
- 从Vicarious起步,最初是AGI实验室,后转向机器人产品
- 为机器人训练计算机视觉模型 - ML领域的第一份工作
- 在Vicarious学习了ML基础设施基础知识
- 加入OpenAI的安全团队,从事代码模型工作
- 看到GPT-3微调后能写代码,意识到AI可以自我改进
- 与安全团队负责人一起离开OpenAI,在Anthropic创立时加入
对AI安全的早期兴趣
- 在GiveWell(评估慈善机构的非营利组织)实习时接触到AI安全概念
- 最初持怀疑态度 - 当时AI安全讨论大多是理论性的
- 模型还不够好,无法构成真正的危险
- 决定从事AI工作:要么安全问题会变得重要,要么就构建有用的AI工具
什么是预训练?
核心概念
- 规模是让AI模型变得更好的关键因素
- 需要一个有海量数据的目标
- 互联网是人类创造的最大单一数据源
- 不需要人工标签 - 从现有文本预测下一个词
工作原理
- 获取文本并预测下一个词
- “The cat” -> 预测接下来是什么
- 每个词都成为新的训练样本
- 从海量数据中获得密集信号
扩展定律
- 随着计算、数据和参数的增加,模型可预测地变得更好
- 更低的损失 = 更好的下一词预测
- 原始扩展定律论文显示了11个数量级的一致趋势
- 这创造了正反馈循环:训练模型 -> 制作有用产品 -> 获得收入 -> 购买更多算力 -> 训练更好的模型
Anthropic早期
小规模起步
- 最初团队约10-12人
- 资金比OpenAI等竞争对手少得多
- GPT-3训练成本约500万美元 - 很多但可以实现
- 使用云服务商的算力
效率作为竞争优势
- “我们的资金比其他人少得多,必须非常高效”
- 当时大多数人使用算力效率不高
- 专注于从可用硬件中获得最大收益
基础设施挑战
- 必须了解芯片的实际物理布局
- 运行聚类算法来识别芯片在哪些机房
- 不同物理位置之间的网络延迟导致问题
- 不断挑战硬件极限
分布式训练方法
关键技术
- 数据并行:将数据分布到多个芯片
- 流水线:跨芯片链接操作
- 操作分片:将操作分布到硬件上
构建自定义基础设施
- 当时没有很好的开源包
- 自己编写all-reduce实现
- 需要修改一切,因为规模超过了Facebook曾达到的
- 不能依赖需要不断修改的外部包
为什么要自建?
- 要达到比现有解决方案支持的更大规模
- 需要完全控制以在每个层面优化
- 可以进行架构更改来简化效率问题
扩展定律与信念
物理学家的视角
- 原始扩展定律论文显示了非常一致的趋势
- 关于趋势是否会继续的争论似乎”毫无意义”
- 11个数量级的数据,争论再多一个点似乎不太可能失败
- “这些人在做什么?他们都错过了大局”
不同实验室的文化
- FAIR(Facebook AI Research)有PhD风格的独立研究文化
- 人们为自己的算力而战,追求个人想法
- 训练大型语言模型需要在基础设施上协作
- 不是可发表的工作 - “我提高了5%的效率”不是论文
- Anthropic的文化专注于协作扩展
团队随时间的演变
早期
- 前3-6个月:试图阅读代码库中的每个PR
- 了解所有部分,可以轻松进行跨领域更改
- 可以通过更改架构发现效率改进
日益专业化
- 随着团队增长,人们成为各个组件的深度专家
- 有些人专注于注意力机制应该如何工作
- 其他人专注于并行策略
- 好处:可以在特定问题上深入研究
权衡
- 专家可能会错过跨组件的优化
- 管理者的角色:确保大局合理
- 需要理解整个系统的人
- 避免单点故障
平衡通才和专家
- 人们对通才vs专家角色有真正的偏好
- 早期员工往往是通才型(创业心态)
- 风险:每个人都做所有事,没有人深入理解任何事
- 专家太多:管理者必须连接一切
- 目标:两种类型的平衡
没有改变的是什么
核心指标
- 仍在推动从第一天起就相同的损失函数
- “损失下降”仍然是唯一重要的OKR
- 可以用相同的指标绘制从第一个模型到今天的进展
- 答案永远是”尽可能低”
反馈循环
- 训练模型 -> 部署 -> 获得收入 -> 购买更多算力 -> 重复
- 这个循环已经持续运行了约5年
- 扩展定律给予信心,进步将继续
关键要点
- 规模很重要:更多的计算、数据和参数可预测地改进模型
- 效率是竞争优势:特别是在资源受限时
- 构建你需要控制的东西:自定义基础设施实现优化
- 平衡团队组成:通才和专家的混合
- 专注于核心指标:损失降低驱动一切