AI-Algorithms

August 22, 2026 · View on GitHub

[toc]

Intro

  • OpenAI 首席科学家 Ilya Sutskever 说过:

    • 数字神经网络和人脑的生物神经网络,在数学原理上是一样的。
  • 大模型最重要的演进方向:

    • 一、世界知识方面如何有效消除幻觉
      • 随着数据规模增大,遇到的新知识比例就越低,在世界知识方面就体现出Scaling law的减缓现象。
    • 二、如何大幅提升复杂逻辑推理能力。
      • 逻辑推理相关数据比例低,更慢。
      • 现在为了提高模型逻辑能力,往往在预训练阶段和Post-training阶段,大幅增加逻辑推理数据占比的原因,且是有成效的。
    • 语言能力已不是问题。
  • Yann LeCun演讲"人类水平的AI"@ Husdon论坛 2024.10

    • 介绍了算法沿袭
    • Moravec's Paradox: AI做不到一些人类很容易做的事情

image-20241019021542281

主流大模型系列

豆包大模型 2.0 (Doubao-Seed-2.0)

参考:豆包大模型2.0发布 | Seed2.0 正式发布

定位:围绕大规模生产环境使用需求优化的大模型系列,支持豆包等上亿用户产品,旨在突破真实世界中的复杂任务。

模型系列

  • Pro:面向深度推理与长链路任务执行场景,全面对标 GPT 5.2 与 Gemini 3 Pro
  • Lite:兼顾性能与成本,综合能力超越上一代主力模型豆包 1.8
  • Mini:面向低时延、高并发与成本敏感场景
  • Code (Doubao-Seed-2.0-Code):专为编程场景打造,与 TRAE 结合使用效果更佳

能力表现

多模态理解

  • 视觉推理、感知能力、空间推理与长上下文理解能力表现尤为突出
  • 数学与视觉推理:在 MathVista、MathVision、MathKangaroo、MathCanvas 等基准上达业界最优
  • 视觉感知:在 VLMsAreBiased、VLMsAreBlind、BabyVision 等基准中取得业界最高分
  • 文档理解:在 ChartQAPro 与 OmniDocBench 1.5 基准上达顶尖模型水准
  • 长上下文:在 DUDE、MMLongBench 等榜单上取得业界最佳分数
  • 视频理解:在 TVBench、TempCompass、MotionBench 等关键测评中处于领先,EgoTempo 基准上超过人类分数
  • 长视频处理:可高效准确处理小时级别长视频,配合 VideoCut 工具进一步提升时长范围和推理精度
  • 流式实时问答:在多个流式实时问答视频基准测试中表现优异,可应用于健身、穿搭等陪伴场景

LLM 与 Agent 能力

  • 科学领域:在 SuperGPQA 上分数超过 GPT 5.2,整体成绩与 Gemini 3 Pro 和 GPT 5.2 相当
  • 数学与推理:在 IMO、CMO 数学奥赛和 ICPC 编程竞赛中获得金牌成绩,超越 Gemini 3 Pro 在 Putnam Bench 上的表现
  • 综合考试:在 HLE-text(人类的最后考试)上取得最高分 54.2 分
  • 工具调用与指令遵循:表现出色
  • 长程任务执行:通过加强长尾领域知识,提升真实世界复杂任务执行能力

成本优势

  • 模型效果与业界顶尖大模型相当,但 token 定价降低了约一个数量级
  • 在大规模推理与长链路生成场景下,成本优势更为关键

应用案例

  • 基于 OpenClaw 框架和豆包 2.0 Pro 模型构建智能客服 Agent,可完成客户对话、拉群求助、预约维修、回访和产品推荐等全流程
  • 豆包 2.0 Code + TRAE:可快速搭建复杂应用,例如 5 轮提示词完成「TRAE 春节小镇 · 马年庙会」互动项目(含 11 位 NPC、AI 游客、烟花/孔明灯实时生成等),相关提示词与素材已开源:https://github.com/Trae-AI/TRAELand

上线情况

  • 豆包 2.0 Pro 已在豆包 App、电脑端和网页版上线,选择「专家」模式即可体验
  • 豆包 2.0 Code 接入 AI 编程产品 TRAE
  • 火山引擎已上线豆包 2.0 系列模型 API 服务
  • 项目主页:https://seed.bytedance.com/zh/seed2

Claude Fable 5 / Mythos 5

参考:Claude Fable 5 and Claude Mythos 5Cloudflare Project Glasswing

定位:Anthropic 2026-06-09 发布的 Mythos-class 模型。Fable 5 是面向一般用户开放的版本;Mythos 5 是同一底层模型,但在部分高风险领域解除 safeguard、走 trusted access 的版本。这里更值得记录的是模型能力谱系,而不是产品权限设计。

LLM 模型能力

  • 长程自主任务能力:官方强调任务越长、越复杂,Fable 5 相对旧 Claude 模型优势越明显;早测案例包括大代码库迁移、长程 coding task、多 agent workflow 和复杂原型构建。
  • 更强 coding 与 token efficiency:在 Cognition FrontierCode 这类高质量生产代码任务上,Fable 5 被描述为 frontier models 中领先;早测反馈集中在 fewer turns、medium effort 下仍有强表现、复杂工程任务能更稳定推进。
  • 复杂知识工作:在金融、交易分析、文档推理、图表 / 表格理解、root-cause analysis、expected-value analysis 等任务上表现突出,说明能力不只是代码生成,而是长文档证据整合与结构化判断。
  • 原生视觉推理:可以从科学图表中抽取精确数字、从截图重建 Web App 代码;官方还给出只依赖原始游戏画面的 Pokémon FireRed 案例,说明模型对视觉状态、导航和任务进展的理解减少了对外部 harness 的依赖。
  • Memory / long-context 使用能力:Fable 5 可以在数百万 token 的长程任务中保持聚焦,并利用自己写下的 notes 改进后续输出;在 Slay the Spire 测试中,persistent file-based memory 对 Fable 的增益显著高于 Opus 4.8。
  • 科学研究与生物推理:Mythos 5 被用于 protein design、bioinformatics、分子生物学假设生成和 genomics research。这里体现的是模型能在工具辅助下完成“选目标、选工具、运行、失败恢复、形成假设”的科研闭环,而不是只回答已有知识。
  • 漏洞利用链与可执行证明:Cloudflare 在 50 多个仓库上的 Project Glasswing 测试显示,Mythos Preview 能把 use-after-free、任意读写、控制流劫持、ROP 等低层攻击 primitive 组合成完整 exploit chain;它还会写 PoC、在 scratch environment 编译运行,并根据失败结果修正假设。能力跃迁不只是“找到更多疑似 bug”,而是把分散线索闭合成可复现证据。
  • 自我检查与反思:早测反馈中多次提到 high effort 下模型会 reflect and validate its own work。这对应 inference-time scaling 的实际价值:额外 thinking token 不只是让答案更长,而是提高长程自主任务中的错误发现和路线修正能力。

模型能力设计判断

Fable / Mythos 这类模型的核心趋势是:LLM 能力正在从“语言、知识、代码”扩展成 reasoning + coding + vision + memory + tool-use + scientific problem solving 的组合能力。评估这类模型时,单一 benchmark 分数不够,要看任务长度、模态复杂度、是否需要外部工具、是否需要跨轮 memory、是否能自我纠错,以及能否在陌生工具 / 真实代码库 / 科研流程中维持稳定行动。

安全研究中的边界同样明显:模型不会像成熟研究员一样主动给出校准后的置信度,容易在“请找漏洞”的目标偏置下返回大量带 possibly / potentially 的猜测;C / C++ 等内存不安全语言又比 Rust 等内存安全语言产生更多误报。PoC 能提高可行动性,但生产使用仍需要独立验证、确定性检查和分层 triage,完整工程闭环见 Cloudflare Vulnerability Harness

算法 Overview

InfiniTensor 大模型概述

  • 计算智能 -> 感知智能 -> 通用智能
  • AGI
    • 头部公司预测3-5年,这个跨度的故事是好故事
  • Note
    • GPT-3.5相比于GPT-3,参数量变化不大,效果差距很大,这是由于微调技术

Agentic Model 所需的模型能力

Agentic Model 就是能支持 Agent 能力的模型。总结来说,Agent 需要模型的这样几种能力:

  • 推理能力,能思考更复杂的任务和规划任务;
  • Coding 编程能力;
  • 多模态能力,尤其是多模态理解能力;
  • 工具使用能力,这和推理、Coding 和多模态能力都相关;
  • 记忆能力,能存储长期的上下文,而且能在处理特定任务时,知道调用哪些适当的上下文。

人工智能发展史

image-20251001190648165

符号智能

image-20251001190833892

专用智能

image-20251001191016712

通用智能

  • 无标注数据+自监督预训练+大模型参数

  • 生成式模型:

    • 判别式,即根据用户和物品的特征预测用户与物品交互的条件概率;而生成式模型预测的是联合概率
    • 判别式关注预测给定条件下的结果,生成式理解所有变量如何共同出现
    • 文本生成:ChatGPT,Gemini
    • 图片生成:Stable Diffusion,DALL-E

    • 视频生成:Sora,Kling

Scaling Law

Scaling Law的前提:统一序列化建模、无监督预训练

image-20251001191246393

image-20251002034314205

Scaling Law

Scaling Law: https://arxiv.org/abs/2001.08361

1代2代3代4代
GPT 系列117M1.5B175B1.7T as rumoured
LIama 系列65B70B70B(3.1代)405B
模型GopherLaMDAChinchillaPaLM
参数280B127B79B540B
  • function form很重要
  • scaling prediction:GPT-4作为了印证
    • image-20251001200843681

截屏2023-11-19 05.00.35

image-20231119050100619

image-20231119050435217

Compute required for training LLMs

Chinchilla: Training Compute-Optimal Large Language Models

  • Google - Chinchilla
    • image-20251001201117306
  • 核心思路:
    • L(N,D)=ANα+BDβ+EL(N,D)=AN^{-\alpha}+BD^{\beta}+E

数据范式演进 (Data Paradigm)

  • 数据受限 (Data Constrained):随着模型规模增长,业界正从"数据无限"进入"数据受限"阶段。
  • 合成数据 (Synthetic Data):Gemini 3 等前沿模型开始大量使用合成数据作为 Scaling 的关键燃料,以突破自然数据枯竭的瓶颈。
    • Chinchilla-optimal
    • loss和参数量呈现对数线性下降
    • D/N20D/N \approx 20
    • 模型undertrained,n和d需要一起增长

emergent ability

利用Scaling Law做科研,小模型的配置推广到大模型

image-20251002023119283

Data Scaling

也参考 「SFT - 指令微调」,有一些手段

高质量数据的价值

image-20251002023555458

缓解数据衰竭 -- "左脚踩右脚"
  • 当前普遍的做法
  • 并不单纯是"左脚踩右脚",过程中,引入了人工校准
缓解数据衰竭 -- Multi-Epoch
  • 数据规模有限时,重复四遍以内基本不影响Scaling Law

image-20251002023412628

其它

  • 量化scaling law,参考Scaling Laws for Precision

  • 参数量 or FLOPS,以MOE为例研究 https://arxiv.org/pdf/2501.12370

    • 核心结论
      • 预训练中优先增加参数而非 FLOPs,最优稀疏度随模型增大趋近 1。
      • 推理时 FLOPs 影响更大,稀疏模型在推理任务需动态增加计算(如思维链提示)。
    • 语言理解类任务依赖参数存储的知识,稀疏模型参数更多优势显著
    • 而推理类任务(如 SQuAD)需要实时计算处理输入,稀疏模型 FLOPs per example 更低,导致推理深度不足,误差比密集模型高 5-10%。

算法创新的方向

  • 现状LLM领域:
    • 提升模型能力:做数据
    • 提升模型效率:做结构

OpenAI 观点:提升数据效率

OpenAI: https://mp.weixin.qq.com/s/M51zPbbNThc8r1WSo7xLoQ

  • 上面的方法只通过线性的增加数据量,就可以有效的提升智能,那我们能不能用这种方法无限的拓展 AI 的智能呢,而无需幂律增加的数据? 不行!因为高级智能模式有这些特点:

    • 种类多

    • 组合多

    • 隐藏的深

    • 发现困难

    • 合成困难

  • 存在巨量的很难发现的,没有被充分表述的,很难合成的智能模式。

    • 如果要 cover 所有的智能模式,需要的数据量还是幂律增长的, scaling-law 依然有效, 数据枯竭的问题还在。
    • Sam 问在没有算法创新的情况下, 这条路线还能走多远? 负责预训练的工程师 Alex Paino 说最多到 GPT 5.5 (数据在 GPT 4.5 的基础上还能再 x10) 。 Alex 及 Selsam都说如果要继续往前走,就需要算法创新,需要追求数据效率,用尽可能少的数量学习到尽可能多的智能模式。 Selsam 说在数据效率上,AI 比人类还差数千倍甚至更多。

密度定律:模型能力密度随时间呈指数级增强

image-20251007010010511

image-20251007010244438

Inference Time Scaling

image-20251007010448786

Literature Review

from InteRecAgent

  • Survey

    • 《Pretrain, prompt, and predict: A systematic survey of prompting methods in natural language processing》
  • LLM capability

    • in-context learning
      • 《few-shot learners》
    • instruction following
    • planning and reasoning
  • alignment

    • 《Training language models to follow instructions with human feedback》
  • leverage LLMs as autonomous agents

    • equipped LLMs with an external memory
    • CoT and ReAct:propose to enhance planning by step-wise reasoning;
    • ToT and GoT:introduce multi-path reasoning to ensure consistency and correctness
    • Self-Refine and Reflexion:lead the LLMs to reflect on errors
    • To possess domain-specific skills,guiding LLMs to use external tools
      • such as a web search engine
      • mathematical tool
      • code interpreters
      • visual models
      • recommender systems

from MindMap

  • LLM应用于生产的局限性
    • Inflexibility.
      • The pre-trained LLMs possess outdated knowledge and are inflexible to parameter updating. Fine-tuning LLMs can be tricky be- cause either collecting high-quality instruction data and building the training pipeline can be costly (Cao et al., 2023), or continually fine- tuning LLMs renders a risk of catastrophic for- getting (Razdaibiedina et al., 2022).
    • Hallucination.
      • LLMs are notoriously known to produce hallucinations with plausible-sounding but wrong outputs (Ji et al., 2023), which causes serious concerns for high-stake applications such as medical diagnosis.
    • Transparency.
      • LLMs are also criticized for their lack of transparency due to the black-box na- ture (Danilevsky et al., 2020). The knowledge is implicitly stored in LLM's parameters, thus infeasible to be validated. Also, the inference process in deep neural networks remains elusive to be interpretable
  • CoT、ToT
    • 挖掘LLM的implicit知识
    • 相应地,MindMap同时挖掘explicit and implicit知识

Attention Is All You Need

Seq2seq

为什么需要 seq2seq 建模

image-20251004025701238

从 RNN 到 Transformer

  • 以RNN为核心的Encoder Decoder有以下几个重要的问题
    • 信息丢失:每次传递乘了系数,丢失前面的信息
    • 无法处理较长句子:RNN 对长期序列依赖关系不稳定,LSTM/GRU 虽一定程度克服长期依赖问题,但无法捕获全局上下文信息。
      • the number of operations required to relate signals from two arbitrary input or output positions grows in the distance between positions, linearly for ConvS2S and logarithmically for ByteNet.
      • RNN是sequence-aligned实现
    • 不能并行计算,对GPU不友好
  • 以上问题,对 从序列到序列(seq2seq)的模型 很重要

image-20251005165225825

Transformer 的目标是 设计全新的、并行的、长期依赖稳定且能捕获全局上下文信息、处理可变长度序列的神经网络架构

image-20241216030117146

从 Machine Translation 的角度理解 Transformer

  • encoder是存储英语原文信息KV
  • decoder逐个生成中文,生成过程中需要以生成的中文为Query,对Encoder高度压缩了的英文KV信息,做cross attn,蒸馏信息并预测生成新query

Intro

  • Intro
    • connect the encoder and decoder through an attention mechanism.
    • Encoder: 映射到另一个语义空间
    • Self-attention, sometimes called intra-attention is an attention mechanism relating different positions of a single sequence in order to compute a representation of the sequence.
  • 公式
    • multi-head self-attention (MSA) + multi-layer perceptron (MLP) blocks
    • image-20241213200148729
  • 模型结构是什么?
    • 过N个注意力层,再过一个full connection
    • Attention(Q,K,V)=softmax(QKTdk)V\text{Attention}(Q, K, V) = \text{softmax}\left(\frac{QK^T}{\sqrt{d_k}}\right)V
      • normalization:dkd_k是head dim(最后一维)
        • 确保注意力机制在不同维度 dk 下都能稳定有效地训练
    • 残差网络
  • 模型参数是什么?
    • 词嵌入向量
      • learnable?
    • 将词嵌入向量转化为q、k、v向量的三个矩阵和bias
      • 线性变换矩阵 WQWKWVW^Q、W^K、W^V
      • 理解Q、K、V:K偏向兴趣和摘要;V偏向原始信息
  • 模型输出是什么?
    • 全连接层的结果,一个长度为全部词汇数量的向量
    • 如何增强随机性:
      • top-k采样

Tokenization 词元化

Intro

  • token是LLM训练推理的最小单元,由tokenizer模型将文本切成token
    • 可能是 1/3 个汉字(因为汉字的UTF-8编码是三个字节,取一个字节)、一个汉字、半个单词等
    • 和模型设计有关:
      • 多语言大模型:汉字拆开
      • 中文大模型比如ChatGLM:一个汉字大概1 token
      • OpenAI的官网上,1 Tokens大概是0.75个英文单词上下(0.5个汉字上下)
    • 和消耗算力有关
      • ->中文大模型更便宜
    • e.g.
      • encoding = encod + ing
      • encoded = encod + ed
      • subword = sub + word
  • Tiktoken
    • 为什么用子词:减少词表的数量
      • 汉字有10万个
如果输入内容是:海南麒麟瓜<br/>
  海, unicode:28023, utf8:b'\xe6\xb5\xb7'<br/>
  南, unicode:21335, utf8:b'\xe5\x8d\x97'<br/>
  麒, unicode:40594, utf8:b'\xe9\xba\x92'<br/>
  麟, unicode:40607, utf8:b'\xe9\xba\x9f'<br/>
  瓜, unicode:29916, utf8:b'\xe7\x93\x9c'<br/><br/>

通过tiktoken处理之后得到的Token序列是:(共11个Token)<br/>
  b'\xe6\xb5\xb7'<br/>
  b'\xe5\x8d\x97'<br/>
  b'\xe9'<br/>
  b'\xba'<br/>
  b'\x92'<br/>
  b'\xe9'<br/>
  b'\xba'<br/>
  b'\x9f'<br/>
  b'\xe7'<br/>
  b'\x93'<br/>
  b'\x9c'<br/><br/>

BPE (Byte Pair Encoding): 常用于文本处理的分词算法

image-20251004030028127

  • 减少未登录词(OOV)的问题

  • Tokenizer的训练与使用

    • 训练语料:BPE算法的词频统计,来源于大模型自身的预训练语料库。在训练一个新模型(如Qwen、Llama)前,开发者会先用其海量的预训练数据(TB级别,包含多语言、代码等)来训练一个专属的Tokenizer。
    • 为何要自己训练
      1. 效率最优化:使Tokenizer能高效地编码其预训练数据中最高频的文本(如中文词语、常见代码片段),从而降低训练和推理成本。
      2. 词表覆盖度:确保词表能覆盖预训练数据中的所有语言和领域。
    • 模型与Tokenizer的绑定:模型学习的是token ID与语义的映射关系。因此,一个预训练好的模型必须使用其在训练时配套的那个专属Tokenizer。使用者在加载模型时,是在加载一个现成的、配对好的Tokenizer,而非重新训练。

Encoder Decoder v.s. Decoder Only

一些思考:

  • 在同等参数下认为 Decoder-Only 架构比 Encoder-Decoder 架构更复杂,因其注意力层输入信息更多对模型能力挑战更大,这种观点有一定合理性
  • Decoder - Only 架构在理论上如果模型能力足够强大,确实有处理长序列并避免明显信息丢失的潜力
  • encoder用于分析,decoder用于生成
    • Decoder 只关注 Encoder 的 最终输出层
    • 下面是一种非标准实现的transformer
      • 标准transformer,几个decoder的输入均为Encoder N的输出
      • image-20250203160834537
  • Encoder Only & Decoder Only & encoder-decoder
    • Decoder Only:将输入拼起来,作为prompt
      • 相比原始transformer,去除了:encoder、decoder中和encoder相连的MSA
      • 转换成了「续写任务」,大部分LLM使用这种架构
      • Decoder-Only模型在参数效率上通常优于Encoder-Decoder模型,因为它不需要同时训练两个模块
    • Encoder Decoder
  • flops对比:
    • N^2 + M^2 + M*N (Encoder-Decoder)

    • (N+M)^2 = N^2 + 2M*N + M^2 (Decoder-Only)

      • cross attention由于其不对称性,在flops上更有优势
    • 并且Encoder-Decoder架构,可以减小Encoder的层数

2025了,如何回答"为什么现在的大模型都是decoder-only的架构?"

  • 表达能力:Decoder-Only模型的自回归注意力矩阵为严格下三角形式并含单位对角线,在理论上保持满秩。Encoder-Decoder结构可能破坏注意力矩阵的满秩性,潜在限制了模型性能上限。
    • 因为Decoder 只关注 Encoder 的 最终输出层
  • 工程角度: Decoder-only 的 KV-Cache 机制天然适配流水线并行和显存优化(如 vLLM 的 PagedAttention)。Megatron-LM、FlashAttention 等底层优化均优先支持因果(Causal)路径。MoE、量化、蒸馏等技术在单解码器结构上更易实现。
  • 预训练难度:每一步都只看左侧信息,任务难度大,因此大模型+大数据下能逼出更通用的表征上限。
  • few-shot/zero-shot:Prompt在所有层都可注入梯度(隐式微调),比 Enc-Dec 两段式更直接。、
  • 隐式位置编码与外推优势:Decoder-Only 将输入输出视为单一连续序列,仅依赖相对位置关系,无需显式对齐编码器-解码器的绝对位置索引。训练后可通过微调或插值轻松扩展上下文窗口(如 LongRoPE),而 Enc-Dec 需处理两套位置系统的兼容性问题。
  • 多模态角度: 主流方案(Gemini/GPT-4o)直接将视觉/音频 tokens 拼接至文本序列,由同一解码器处理,实现"早融合"的工程最优解。
  • 轨迹依赖:openai率先验证了该架构的训练方法和scaling law,后来者鉴于时间和计算成本,自然不愿意做太多结构上的大改动,就继续沿用decoder-only架构,迭代 MoE、长上下文、多模态。

Encoder / Decoder

从 classification 的角度理解 Attention

image-20250511160613431

多头自注意力 MSA

  • 从模型复杂度的角度:假设超参对效果贡献相同,优先让模型更复杂,利于Scalable,更晚遇到天花板

    • "多头"比"单头"复杂,是比较直观的

    • "多头" v.s. overlap,类似于推荐系统中的share emb,不确定哪种"更复杂"

      • overlap利于提取特征的局部细节,对于语言类模型,不知道有没有用

  • 动机:缓解全局attention的信息丢失
  • The Transformer follows this overall architecture using stacked self-attention and point-wise, fully connected layers for both the encoder and decoder, shown in the left and right halves of Figure 1
    • 左边encoder,右边decoder
      • Encoder: 自注意力
      • Decoder:Q用outputs embedding做masked attention后的结果,K、V用encoder结果
      • 表征向量512维
    • 自注意力机制:Q(输入矩阵)、K(字典)、V
      • K用来计算依赖关系
      • 用1/(dk)^(1/2) scale了一下QK的乘法,可能是为了防止gradient太小
  • Multi-head attention: 多头自注意力机制
    • 多头注意力机制(Multi - Head Attention)的计算表达式为: MultiHead(Q,K,V)=Concat(head1,,headh)WO\text{MultiHead}(Q, K, V) = \text{Concat}(\text{head}_1, \ldots, \text{head}_h)W_O
      • 其中每个头的计算公式为: headi=Attention(QWQi,KWKi,VWVi)\text{head}_i = \text{Attention}(QW_{Q_i}, KW_{K_i}, VW_{V_i})
    • 自注意力和CNN的辨析 https://www.mstx.cn/pytorch/209.html
      • 相似性:信息提取机制、并行、可堆叠
      • 区别:感受野的固定性和灵活性、局部性和全局性、计算复杂度、空间维度与序列维度

image-20231025203852956

image-20241216030854804

image-20241216030905082

  • 自注意力:
    • 本质上是信息的聚合
    • 计算复杂度:O(N^2)
    • 经典的transformer:6层
    • GPT: 12层
    • GPT-2: 32层
    • GPT-3: 96层
    • GPT-4、llama3:120层
    • Q、K、V不同的理解:Q、K、V通过不同的线性变换从同一输入序列中生成,各自承担着不同的作用,共同实现了模型对序列中不同位置之间关系的捕捉。
为什么 multi-head
  • 《Attn is all you need》 "Multi-head attention allows the model to jointly attend to information from different representation subspaces at different positions. With a single attention head, averaging inhibits this."
  • But it also is a formidable computational simplifications: The heads operate fully independently, so computing them is (like batch) "embarrassingly parallel"
    • head dim是性能的一个限制因素 --> GPU Kernel,SM并行计算
Q和KV的head num可为倍数关系(GQA)

image-20251004230520582

Self-Attn 的数学特点
  • QA和KA为每行相等的常量时,有以下两个特性:

    • VB主体:self-attn值不受影响
    • VA常量:由于softmax具有归一化的特点,这里具有不变性
  • O=softmax(QKTτ+mask)V=softmax([QA,QB][KA,KB]Tτ+mask)[VA,VB]=softmax(QAKAT+QBKBTτ+mask)[VA,VB]=softmax(QBKBTτ+mask)[VA,VB]=[VA,softmax(QBKBTτ+mask)VB]\begin{array}{rcl} O &=& softmax(\frac{QK^T}{\tau} + mask)V \\\\ &=& softmax({\frac{[{Q_A}, Q_B] [{K_A}, K_B]^T}{\tau}}+ mask)[{V_A}, V_B] \\\\ &=& softmax({\frac{{Q_A K_A^T} + Q_B K_B^T}{\tau}}+ mask)[{V_A}, V_B] \\\\ &=& softmax(\frac{Q_B K_B^T}{\tau}+ mask)[{V_A}, V_B] \\\\ &=& [{V_A}, softmax(\frac{Q_B K_B^T}{\tau}+ mask)V_B] \end{array}

Self-Attn 是低通滤波器

ANTI-OVERSMOOTHING IN DEEP VISION TRANSFORMERS VIA THE FOURIER DOMAIN ANALYSIS: FROM THEORY TO PRACTICEhttps://arxiv.org/pdf/2203.05962

  • 视觉 Transformer(ViT)在深度增加时因注意力坍塌和补丁均匀性导致性能饱和,本文通过傅里叶分析建立理论框架,证明自注意力机制本质上是低通滤波器,深度增加会使特征图仅保留直流(DC)分量。为此提出 AttnScale 和 FeatScale 两种技术:前者将自注意力块分解为低通和高通分量并重新缩放组合为全通滤波器,后者对不同频带特征图重新加权以增强高频信号。两者均无超参数且高效,插入多种 ViT 变体后,使 DeiT、CaiT 和 Swin-Transformer 性能分别提升最高 1.1%、0.6% 和 0.5%,参数开销极小。

  • self-attn是低通滤波器

    • 定理与推论
      • 定理 1:自注意力矩阵是低通滤波器,随层数增加高频分量消失。
      • 推论 2:不同层自注意力矩阵的组合仍为低通滤波器。
      • 定理 3:给出自注意力对高频分量的抑制速率上界。
    • 现有机制的作用
      • 多头注意力、残差连接和前馈网络(FFN)可缓解但无法根除低通问题。
      • 残差连接能防止高频分量衰减至零,但无法单独提升高频信息。
    • image-20250605194854393
    • 本质上是softmax是低通滤波器
    • image-20250605200742218
Massive Values in MSA

Massive Values in Self-Attention Modules are the Key to Contextual Knowledge Understanding

Quiet Attention

https://www.evanmiller.org/attention-is-off-by-one.html

https://github.com/kyegomez/AttentionIsOFFByOne

image-20250606172056691

softcapping

softcapping

  • image-20250924141137584

Decoder

image-20251004224418837

广义的 decoder

  • 也参考「深度学习推荐系统--VQ-VAE」

Masked Softmax 因果掩码机制

  • masked multi-head attention

    • 保证输出对输入的感知序列不会超出长度:防止在训练过程中模型看到未来的信息,确保预测是基于之前的输出

    • 对 QK^T 做mask

    • 注意力矩阵:下三角非零

    • A = A + M,M的某些元素是 -\infty

交叉多头注意力层

  • Q来自Decoder:考虑已经生成的内容
  • K、V来自Encoder:考虑上下文
    • 传统 Transformer Decoder 的局限性:传统 Transformer Decoder 主要依靠输入的 K、V 与 Q 计算注意力,进而生成输出。当输入短,K、V 提供的信息不足,注意力机制可聚焦的范围窄,解码器难以跳出有限信息的限制,导致预测结果单一。

KV Cache的可行性

  • 能否直接更新历史 KV?
    • 理论上,你可以设计一种机制,在生成第 t 个 token 时,不仅计算 Q_t , K_t , V_t ,还去修改缓存中 K_1...K_{t-1} 和 V_1...V_{t-1} 的值。
  • 为什么通常不这样做?
  1. 破坏 KV Cache 的核心优势 :如果每一步都要更新所有历史 K/V,推理成本将急剧增加,从 O(N)(N 为序列长度,使用 Cache)变回 O(N^2)(每次都重新计算或更新所有历史 K/V),失去了 Transformer 推理效率的关键优化。
  2. 改变了注意力机制的含义 :标准的自注意力机制假设一个 token 的 K 和 V 代表其在 那个时间点 的上下文表示。基于 未来 的 token 来修改 过去 token 的 K/V 表示,改变了这种前向因果关系,使得模型结构和信息流变得复杂。这更像是双向模型(如 BERT)在编码整个序列时做的事情,而不是自回归生成模型逐词生成时的工作方式。
  3. 实现复杂且收益不明确 :设计一个有效且稳定的更新历史 K/V 的机制会非常复杂,并且不清楚这样做是否能带来足够的好处来抵消巨大的计算成本和复杂性增加。

FFN

MLP

  • 提升网络表达力
  • 非线性
  • 融合多头特征

Native FFN (PositionWiseFeedForward)

GLU Variants Improve Transformer https://arxiv.org/pdf/2002.05202

image-20250703152649327

SwiGLU

  • FFN_SwiGLU(x) = (Swish(xW₁) ⊙ xW₃)W₂

  • 一种 动态的、依赖于输入内容的特征选择机制

image-20251002161617087

激活函数的稀疏化探索 -- ReLUfication、ProSparse、ReLU^2
  • 图中s1:Swish替换为ReLU
  • 图中s2: 也在layernorm后插入额外的relu层

image-20251003003146309

  • ProSparse
    • MiniCPM-S

image-20251003003248893

  • ReLU^2
    • image-20251003003357703
    • token维度,相邻token复用神经元;流量维度,神经元的共现情况
      • image-20251003003547686

Layer Normalization

  • The LayerNorm operator is a way to improve the performance of sequential models (e.g., Transformers) or neural networks with small batch size
    • 对序列中每个 token 的 hidden embedding(如 768 维)做 LN
  • LayerNorm 的核心逻辑是样本内归一化:在单个样本 / token 的特征空间内求均值和方差,保持样本独立,避免跨样本统计量干扰。
维度BatchNormLayerNorm
统计范围同一特征 / channel 在 batch 内求均值方差;表格直觉是“按列归一化”单个样本 / token 内所有 hidden 维求均值方差;表格直觉是“按行归一化”
依赖依赖 batch 统计量;训练和推理不同,推理用 running mean / variance不依赖 batch size;训练和推理行为基本一致
适用CNN / 图像等 batch 统计稳定、channel 语义固定的场景RNN / Transformer / 文本语音等序列模型,小 batch 或动态长度更稳
直觉利用同一批样本校准每个特征维度让每个样本自己的表示尺度稳定
  • 对 Transformer 张量 [B, T, H]:LayerNorm 通常对每个 (B, T) 位置上的 H 维 hidden state 做归一化,而不是跨 batch 或跨 token 求统计量。
Pre-LN
  • 将归一化层放在子层(Attention 或 FFN) 之前 的结构被称为 Pre-LN (Pre-Layer Normalization) 。
  • 主要原因和优点如下:
    1. 训练稳定性 :这是采用 Pre-LN 最主要的原因。在原始的 Post-LN 结构 (Input -> Sublayer -> Add -> LayerNorm) 中,随着网络层数加深,每一层的输出在累加(Add 操作)后才进行归一化,可能导致梯度在反向传播时出现剧烈变化(梯度消失或爆炸),使得训练过程不稳定,尤其是在模型很深的时候。Pre-LN 结构 (Input -> LayerNorm -> Sublayer -> Add) 通过在每个子层的输入处进行归一化,稳定了传递给子层的激活值范围,从而也稳定了反向传播时的梯度流。这使得训练过程更加平滑,不易发散。
      • 反向传播路径 : d(Output) -> d(Add) -> [d(Sublayer) -> d(LayerNorm)] AND [d(Input)]
      • 在 Add 节点,梯度"兵分两路":
        • 路径一(主干道/残差连接) : 梯度 直接、原封不动地 流向 Input 。这是一个恒等映射(Identity Path),梯度值乘以1。这是最关键的一点,它为梯度提供了一条 畅通无阻、无任何缩放 的"高速公路",可以直接回传到网络的更深层。
        • 路径二(分支) : 另一份梯度流向 Sublayer ,然后穿过 Sublayer 的反向传播,再穿过 LayerNorm 的反向传播,最后这个经过计算和缩放的梯度也作用于 Input 。
      • 结论(Pre-LN) :Pre-LN 之所以稳定,是因为 主干道是"干净"的 。无论分支(路径二)上的 Sublayer 和 LayerNorm 产生了多大的梯度,它们只是作为一部分"增量"被 加到 主干道的梯度上,而不会改变主干道梯度本身的直接传导。
    2. 减少对学习率 Warmup 的依赖
    3. 更快的收敛(有时)
Post-LN 的问题

Input -> Sublayer -> Add -> LayerNorm

  • 反向阶段,LayerNorm的梯度较大,再基于残差层传入Sublayer,梯度大小不可控
最后一层加 LN
  • an additional layer normalization was added after the final selfattention block 【GPT-2】
RMS Norm

image-20251004224626759

Google 去除 LN 的尝试

参数初始化策略

Position Encoding

https://arxiv.org/pdf/1705.03122

  • 为什么引入?

    • MSA的计算,改变Q和K的词元位置,计算结果不变,"invariant to position"
  • 绝对位置编码:

    • Convolutional Sequence to Sequence Learning
    • 正弦-余弦编码
  • 相对位置编码:

    • 作用于自注意力机制

正弦-余弦编码

在Transformer架构中,由于其核心组件(如多头自注意力机制)本身不具备捕捉序列中元素位置信息的能力,所以需要额外的位置编码来为模型提供位置信息。正弦 - 余弦编码通过使用正弦和余弦函数来生成位置编码向量,其基本思想是利用不同频率的正弦和余弦波来表示不同的位置。

对于一个长度为 LL、维度为 dd 的序列,位置编码 PEPE 是一个 L×dL\times d 的矩阵,其中第 pospos 个位置、第 ii 个维度的编码值计算公式如下:

ii 为偶数时: PE(pos,2i)=sin(pos100002id)PE_{(pos, 2i)} = \sin\left(\frac{pos}{10000^{\frac{2i}{d}}}\right)

ii 为奇数时: PE(pos,2i+1)=cos(pos100002id)PE_{(pos, 2i + 1)} = \cos\left(\frac{pos}{10000^{\frac{2i}{d}}}\right)

其中,pospos 表示位置(范围从 0 到 L1L - 1),ii 表示维度(范围从 0 到 d21\frac{d}{2}-1),dd 是位置编码向量的维度。

  • 提供位置信息:通过正弦 - 余弦编码,模型能够区分序列中不同位置的元素,从而学习到元素之间的相对位置关系。这对于处理序列数据(如自然语言、时间序列等)至关重要,因为元素的顺序往往携带了重要的语义信息。
  • 线性可学习:正弦 - 余弦编码具有一定的线性特性,使得模型可以通过线性变换来学习位置信息,从而提高模型的学习效率。
  • 外推性:由于正弦和余弦函数的周期性,正弦 - 余弦编码具有较好的外推性,即模型可以处理比训练时更长的序列。
    • may allow the model to extrapolate to sequence lengths longer than the ones encountered during training.

优点

  • 无需学习
  • 相对位置信息:比如B在A后面N个位置,C在B后面N个位置
  • 计算高效

缺点

  • 固定模式
  • 缺乏语义信息:只提供了位置信息,不包含元素的语义信息,对于一些需要结合位置和语义信息的任务,可能需要与其他编码方式结合使用。

RoPE等参考「Long Context」

用户感知的位置编码

MMGRec

  • pQu(j)=ejWuQ;WuQ=MLP(hu)WQpKu(j)=ejWuK;WuK=MLP(hu)WKpVu(j)=ejWuV;WuV=MLP(hu)WV\begin{array}{rcl} \mathbf{p}_Q^u(j) = \mathbf{e}_j \mathbf{W}_u^Q &;& \mathbf{W}_u^Q = MLP(\mathbf{h}_u) \mathbf{W}^Q\\\\ \mathbf{p}_K^u(j) = \mathbf{e}_j \mathbf{W}_u^K &;& \mathbf{W}_u^K = MLP(\mathbf{h}_u) \mathbf{W}^K\\\\ \mathbf{p}_V^u(j) = \mathbf{e}_j \mathbf{W}_u^V &;& \mathbf{W}_u^V = MLP(\mathbf{h}_u) \mathbf{W}^V \end{array}
  • score=softmax(QKT+PEQKD), where PEQK=pQu(j)pKu(j)Tscore = softmax \left( \frac{QK^T + PE_{QK}}{\sqrt{D}} \right), \text{ where } PE_{QK} = \mathbf{p}_Q^u(j)\mathbf{p}_K^u(j)^T

Output Embedding 映射到预测空间

  • 线性层、词表数量
    • 隐藏层的映射目的:神经网络中所有隐藏层的映射,本质是不断在多个超空间中对特征进行相互映射。
    • 表示学习假设:表示学习假定存在一个超空间,在这个超空间里能够找到一个超平面,将空间中的目标向量区分开来。
  • Tie/shared word embedding: share the same weight matrix between the two embedding layers and the pre-softmax linear transformation
  • 可以只对需要输出的logits进行这步计算

训练策略

序列打包(Sequence Packing)

  • 动机:将变长语料构造成固定长度训练序列,最大化有效上下文与吞吐,最小化无效 padding。
  • 训练目标:
    • minθ  Ex1:LSt=1Llogpθ(xtx<t)\min_{\theta}\; \mathbb{E}_{x_{1:L}\sim \mathcal{S}} \sum_{t=1}^{L} -\log p_{\theta}(x_t \mid x_{<t})
    • EOS 用作语义边界;是否允许跨文档注意由策略决定。
  • GPT 系列的默认实践:
    • 文档流式拼接(Streaming with EOS),截断为固定长度:GPT-1(≈512)、GPT-2(1024)、GPT-3(2048)。
    • 长度分桶(Bucketing)与最小填充,降低 pad 比例,提升吞吐。
    • 一般允许跨文档注意,无显式跨样本掩码。
  • 现代工程扩展:
    • 跨样本打包 + 注意力掩码(packed samples with masks):将多条短样本拼接为一条,并用掩码阻断跨样本注意,兼顾"无填充"与样本独立性。
    • Token-budget 动态批(token-based batching):以固定 token 预算组织 batch,样本数随长度变化,提高设备利用率。
  • 影响与权衡(Zhao et al., 2024):
    • 过度碎片化削弱长程依赖;保持文档连续性与明确 EOS 边界通常更优。
    • 当任务对边界敏感时,打包+掩码更稳;一般预训练场景允许跨文档注意更有利。
    • 调参关注"有效 token 利用率"(非 padding token 占比)与"跨文档比率"。
  • 实操建议:
    • 默认采用"流式拼接 + EOS + 长度分桶/Token-budget 动态批"。
    • 需要严格样本独立时,使用"打包 + 掩码"。
  • 参考:

Label Smoothing

  • During training, we employed label smoothing of value ϵls=0.1ϵls=0.1 (cite). This hurts perplexity, as the model learns to be more unsure, but improves accuracy and BLEU score.
  • label [2,1,0,3,3]
    • image-20250207130816676
  • 实现参考「code-reading-the-annotated-transformer」

推理采样

Intro

  • 推理:
    • <sos>(start of sentence)
    • 不断过decoder
    • 直到生成eos
  • 推理和训练的区别
    • 推理阶段的操作和训练阶段的解码器操作类似,但是训练阶段有目标序列的真实值作为输入来计算损失并进行反向传播训练,而推理阶段是根据之前生成的单词不断生成新的单词。
    • 在训练时,解码器的输入是已知的目标序列,在推理时,解码器的输入是逐步生成的单词序列。

Argmax和随机采样

image-20251005010204784

temperature、top-k、top-p
  • top-p采样:也称为nucleus sampling

image-20251005011425211

image-20251005011715687

  • 基本思想:为了克服贪心搜索的短视,Beam Search 在每一步都会保留 k (beam width) 个最可能的候选序列。在下一步,它会基于这 k 个序列,分别生成所有可能的下一个 token,并从中再次选出总概率最高的 k 个新序列。这个过程持续进行,直到生成结束符或达到最大长度。

  • 优点:通过保留多个候选,它有更大的机会找到全局最优或接近最优的序列,生成质量通常高于贪心搜索。

  • 缺点:需要更大的计算和内存开销,因为需要同时维护 k 个序列的状态,尤其是 KV Cache 的存储。

  • Trie-Based Beam Search (基于前缀树的束搜索优化)

    • Efficient Beam Search for LLMs Using Trie-Based Decoding
    • 动机:传统的 Beam Search 为每个候选序列(beam)都独立存储一份 KV Cache,当多个 beam 共享相同的前缀时,这会造成大量的内存冗余。
    • 核心方法:该方法使用 Trie (前缀树) 结构来统一管理所有 beam 的 KV Cache。共享相同前缀的 beam 在 Trie 树中会共享同一条路径,从而共享同一份 KV Cache。只有当 beam 发生分叉时,才会在树上创建新的节点来存储差异部分。
    • 效果:在不牺牲生成质量的前提下,该方法能节省 4-8 倍的内存,并带来最高 2.4 倍的解码速度提升,极大地优化了 Beam Search 的推理效率。

局限性

实验

image-20250205164614941

Transformer的改进

【InfiniTensor】大咖课-韩旭-知识密度牵引下的大模型高效计算

KV压缩

Intro

  • KV cache大小

    • image-20251005025202463
  • 优化n_head

    • MQA、GQA

MQA

  • MQA

    • image-20250601014800391
  • 从MHA到MQA的up-training方法【GQA paper】

image-20250502141431027

GQA

GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpoints

主要动机是优化推理时加载KV cache的memory bandwidth开销

image-20250502141600078

  • 降低了计算量 :

    • 在 MHA 中,每个 Query Head 都有自己独立的 Key Head 和 Value Head。
    • 在 GQA 中,多个 Query Head 被分成组, 同一组内的 Query Head 共享同一对 Key Head 和 Value Head 。
    • 这意味着计算 Key 和 Value 的投影以及后续的 Attention Score 计算量减少了,因为 Key 和 Value 的 Head 数量远少于 Query Head 的数量(介于 MHA 和 MQA 之间)。
  • 大幅减少了 KV Cache 的大小 :

    • 在自回归生成(Inference)过程中,需要缓存过去所有 token 的 Key 和 Value 状态(即 KV Cache),这部分显存占用非常大,尤其是在处理长上下文时。
    • 由于 GQA 共享 K/V Head,需要缓存的 K/V 张量数量大大减少(减少的比例等于分组的大小 G)。例如,如果有 8 个 Query Head,分为 2 组(G=4),那么 K/V Head 的数量就从 8 对减少到了 2 对,KV Cache 的大小也相应地减少为原来的 1/4。
    • 这显著降低了推理时的显存占用,使得在有限的硬件上可以运行更大的模型或处理更长的序列。
  • 提高了推理速度 :

    • 减少 KV Cache 不仅节省显存,更重要的是 减少了内存带宽的压力 。在推理时,从显存加载巨大的 KV Cache 是一个主要的速度瓶颈。GQA 通过减小 KV Cache 大小,显著加快了这部分数据的读取速度。
    • 计算量的减少也对推理速度有一定贡献。

MLA

见 DeepSeek-V3 章节

「稀疏注意力」章节

Q压缩

Perceiver

Yannic Kilcher 论文精读:https://www.youtube.com/watch?v=P_xeshTnPZg&t=4s

结合代码分析:https://zhuanlan.zhihu.com/p/360773327

动机
  • 同时处理多模态是趋势:Biological systems perceive the world by simultaneously processing high-dimensional inputs from modalities as diverse as vision, audition, touch, proprioception, etc.
  • 过往CV和语音模型设计,局限于单模态的处理:The perception models used in deep learning on the other hand are designed for individual modalities, often relying on domain-specific assumptions such as the local grid structures exploited by virtually all existing vision models. These priors introduce helpful inductive biases, but also lock models to individual modalities.
  • Perceiver - a model that builds upon Transformers and hence makes few architectural assumptions about the relationship between its inputs, but that also scales to hundreds of thousands of inputs, like ConvNets.
    • 算法考虑:同时处理多模态
    • 工程考虑:
      • NLP:几千token; CV:50k~2242224^{2} token
      • 聚类压缩Query:the model leverages an asymmetric attention mechanism to iteratively distill inputs into a tight latent bottleneck, allowing it to scale to handle very large inputs.
Model 设计
  • 角度一:encoder-decoder --> Perceiver <-- Decoder-only
    • encoder-decoder --> Perceiver:去除对原始序列的encoder,减少Flops
    • Perceiver <-- Decoder-only:压缩query,利用cross-attn,减少Flops
  • 角度二:RNN的角度理解
    • K、V是序列输入
    • Q是latent state
    • K、V是输入,将信息重复蒸馏进Q中
  • 角度三:将Transformer放平
    • 相同点:大的byte array通过cross attn蒸馏小的latent array

    • 不同点:

      • latent array(output sequence)是随机初始化的,大小可以随便控制;而原来的面向seq-to-seq学习的transformer中的output sequence是来自目标语言的表示层;
      • transformer中对output sequence是先self-attention,然后cross-attention;而perceiver中则相反,是先cross-attention,然后再进行若干次self-attention。
    • image-20250627151342683
  1. 核心机制
    • 非对称交叉注意力:查询(Q)来自可学习的低维潜在单元(N=512),键(K)和值(V)来自输入数据(M≥50,000),将复杂度降至 O (MN)。
      • 初始Q的选取:
        • 可学习初始化: KK 个的 Variables 作为聚类中心(K<<LK << L
          • 类似Q-Former
        • hard方式初始化: 取Seq的前K个token,包括:1)k1个global token;2)序列中前K-k1个token
      • 利用 cross attention 将序列信息蒸馏到可学习的 Variables 中,实现自适应聚类
      • 利用 self attention 捕捉聚类中心之间的高阶交互关系
      • 交错进行 2、3 两个步骤将模型堆叠至多层
    • 迭代蒸馏:交替使用交叉注意力(提取输入特征)和潜在 Transformer 自注意力(处理低维表示),如图 1 所示,通过 8 次迭代逐步聚焦关键信息。
      • image-20250607013213140
    • 权重共享:后续交叉注意力模块共享参数,减少过拟合,参数数量减少约 10 倍(如 ImageNet 模型从 326.2M 降至 44.9M)。
  2. 位置编码
    • 傅里叶特征:使用高频正弦余弦编码空间坐标(如 2D 图像的 (x,y)),支持高分辨率表示(如 224×224 图像用 64 频带)。
    • 模态标识:为多模态输入添加模态特定编码(如视频 + 音频时用 4 维嵌入区分)。
  3. 进一步改造
    • 压缩到极致即为 attention sink 作为输入 ,做单步解码
结论
  • Scaling law

    • image-20250626171040462
  • weight sharing减少参数量,缓解过拟合

    • image-20250626165611055
  • cross-attn和self-attn interleaved效果好

    • image-20250626165801217

QKV压缩

Token Merge

LONGER: Scaling Up Long Sequence Modeling in Industrial Recommenders

  • 对block内做concat

Trans in trans

https://arxiv.org/pdf/2103.00112

  • 在CV中应用,主要解决image token化时, patch过大(16X16)内部信息没有得到较好提取的问题。解决方法是在16X16的patch分解出4X4的patch, 先做一次transformer, 再在外层用16X16的token做trans.

image-20250606173644337

稀疏注意力

Intro: Sparse Transformer 和 Longformer

  • Sparse Transformer:通过滑动窗口、周期性稀疏注意力进行加速。
  • Longformer:通过跳跃滑动窗口、周期性稀疏注意力、局部全局注意力进行加速。

image-20251007014852428

Sliding Window Attn -- Mistral-7B

本质上是为 Transformer 模型注入 "局部性优先" 归纳偏置 (Inductive Bias)

  • GQA + Sliding Window Attn + Rolling Buffer Cache
    • 减少计算和KV存储

image-20250503011135579

Streaming LLM: Attention Sink (Global Token)

EFFICIENT STREAMING LANGUAGE MODELS WITH ATTENTION SINKS (streaming-LLM)

节点间信息传播的中心,前几个token起到锚点的作用

global token可缓解稀疏注意力中对长距离依赖性建模能力的退化 paper1paper2paper3

image-20250606171300149

  1. 注意力汇聚点(Attention Sink)
    • 发现模型对初始 tokens 分配大量注意力,即使其语义无关(如 Llama-2-7B 深层头部对初始 token 注意力占比超 50%)
    • 原因:Softmax 要求注意力分数和为 1,初始 tokens 因自回归特性被所有后续 tokens 可见,易被训练为汇聚点
  2. StreamingLLM 框架
    • 核心设计:保留 4 个初始 tokens 的 KV 作为汇聚点,结合滑动窗口 KV 缓存(如 4+1020 配置)
    • 技术细节
      • 缓存内重新分配位置编码(如当前缓存 tokens [0,1,2,3,6,7,8] 解码时位置设为 0-7)
      • 兼容 RoPE(缓存 Keys 后应用旋转变换)和 ALiBi(连续线性偏置)
    • 预训练优化:添加 Learnable Sink Token 作为专用汇聚点,160M 参数模型实验显示仅需 1 个该 token 即可稳定性能

四、应用与局限

  1. 适用场景:多轮对话、短文档 QA 等依赖近期上下文的流式任务,已被 NVIDIA TensorRT-LLM 等框架采用
  2. 局限性
    • 不扩展模型上下文长度,依赖缓存内信息(如 StreamEval 中查询距离超缓存时准确率降为 0)
    • 长文档 QA 等需长期记忆的任务表现不及截断基线

image-20250606171351749

MInference: StreamingLLM + InfLLM + vertical-slash
  • vertical slash:
    • 归纳偏置:在一个局部的Query块内,认为所有Query向量的注意力分布是相似的,因此用这个块里最后一个Query的注意力分布来代表整个块的注意力分布。

image-20251007134921672

MoBA Attn (Moonshot AI): 压缩 - hard选择

https://arxiv.org/pdf/2502.13189

MOBA: MIXTURE OF BLOCK ATTENTION FOR LONG-CONTEXT LLMS

  • selection
    • 对block内做mean pooling

image-20251005232237411

image-20251005232321586

类似工作:InfLLM v2 / MiniCPM4

image-20251007140013419

image-20251007140911453

[DeepSeek] NSA (native sparse attn) 压缩 - soft选择 - 滑窗

DeepSeek的优势在于基本是最早用NSA方案做了pre-train,预训练做稀疏训练

面向 Inference Time Scaling

  • 架构概览

    • 压缩 -- 选择 -- 滑窗
    • image-20251005145437139
    • 压缩:8倍,比较激进
    • 选择:
      • 图右上角的mask比较有意思,对于尾部token的处理
      • Gated Output
  • kernel design

    • image-20251005150446270
    • 挑战是KV稀疏,SRAM放什么数据需要设计
    • 解决方案:用FA-2的思路,Query放在SRAM
      • 前提:使用了GQA
  • 效果:

    • image-20251005150728341
    • image-20251005145214936
分块压缩:InfLLM

image-20251007134505990

模型化:选取代表元

Explicit Sparse Transformer

image-20251007015647275

非Transformer架构研究

todo: 快速权重 using fast weights to attend to the recent past

Intro

RNN 长上下文记忆能力弱 --> Transformer架构

Transformer长上下文性能差 --> 新一代RNN架构

image-20251007141002959

架构设计者特点链接
TransformerGoogle最流行,几乎所有大模型都用它OpenAI 的代码
RWKVPENG Bo可并行训练,推理性能极佳,适合在端侧使用官网RWKV 5 训练代码
MambaCMU & Princeton University性能更佳,尤其适合长文本生成GitHub
  • 目前只有 transformer 被证明了符合 scaling-law。
    • 收效甚微
    • 这些新框架,主要用在端侧大模型
    • 大公司追求效果极致的好
  • RWKV、Mamba:线性transformer
    • mamba:选择性SSM架构
  • Additive Attention https://arxiv.org/abs/1409.0473

Linear Attn 基础 - Transformer without softmax == RNN

  • V=(φ(Q)φ(K)T)VV' = (φ(Q)φ(K)^T)V
    • 复杂度从序列长度N的平方,降到了线性
    • 可以把 (φ(K)^T * V) 看作一个 全局的"状态"或"记忆" 。它把所有 Key 和 Value 的信息压缩成了一个小矩阵。
    • 在自回归生成(一个一个token地生成)的场景下,这个"状态"可以迭代更新

image-20251007015959086

评价Linear Attn:联想记忆和状态追踪能力

image-20251005170207628

(RWKV7)

第二代RNN:提升训练质量和效率

记忆压缩

image-20251007141822655

第三代RNN:提升模型记忆能力

快慢记忆融合机制

RWKV

读法:raku

image-20251005165507125

  • 核心差异:Time Mixing
    • R:使用多少信息量
    • W:信息量的重要性

image-20251005170437615

Time Mixing

image-20251005165713281

RWKV-7

image-20251005165755159

image-20251005170039565

image-20251005170356714

MoE 混合专家模型

https://huggingface.co/blog/moe

思路:扩参数量,保Flops不变

Intro

MoE介绍

  • 门控网络+专家网络
  • GPT-3 1750亿参数
  • GPT-4 1.8万亿参数
    • 16个专家网络
    • 运行时只跑2个专家网络
    • 相比GPT-3.5更像人脑

image-20251003005026169

image-20251003005055185

  • 发展趋势
    • image-20251003010400486

MoE和人脑的类比

image-20251007012257695

  • 高效性:MoE的本质特点
  • 可复用性:模块化复用与组合正是MoE模型能够掌握和处理海量、多样化知识与任务的理论基础
    • 人脑:模块复用和组合
    • MoE:不同层的gating不一样,相当于不同层激活的专家可能对于不同任务有复杂的组合。
  • 可解释性
    • 人脑 :神经科学家通过**功能性磁共振成像(fMRI)等技术,观察人类在执行特定任务时大脑的哪些区域(模块)被激活,从而推断这些区域的功能。例如,当一个人说话时,如果布罗卡区(Broca's area)**被点亮,我们就能理解它在语言生成中的关键作用。
    • MoE:虽然模型的完全可解释性仍是一个前沿课题,但相比于密集的"黑箱"模型,MoE在可解释性上提供了重要的突破口。
      • 通过分析和追踪哪些类型的输入token总是被路由到某一个特定的专家,研究人员可以推断出这个专家的"专长"。例如,已经有研究发现,在多语言模型中,存在专门处理特定语种(如德语、日语)的专家,也存在专门处理代码、JSON格式或科学文献的专家。

LLM神经元的稀疏激活特性

FFN
  • 核心:激活函数输出的激活值,具备稀疏性

image-20251007013022728

Attn也具备

image-20251002163423266

从FFN到MoE:FFN模块化

模块化

image-20251002162330247

  • 模块化性质的早期涌现,先构建模块化,再构建神经元功能

image-20251002163239043

image-20251007013206293

神经元特异化:能力&语言&情感神经元

  • image-20251003002100921

  • image-20251003002248264

  • 情感神经元

    • 情感神经元在所有层 --> 需要微调来实现角色扮演的大模型
    • image-20251003002341416

神经元激活的稀疏性 -- MoeFication

Question: paper如何去判断神经元功能分布相似的,需要批量输入+统计?

image-20251002162209654

image-20251007013515941

稀疏稠密协同训练

可能被淘汰了,现在LLM一开始就是稀疏结构

image-20251003003642702

SparseMoE

  • 每个token分配到Gate分数最高的k个Experts上进行计算
  • 问题:
    • load balance
    • 访存bound:Expert parallelism

Load Balance

auxiliary loss -- DeepSeek-V2

image-20251003010847599

随机路由

image-20251003005507992

auxiliary-loss-free -- DeepSeek-V3
  • For MoE models, an unbalanced expert load will lead to routing collapse (Shazeer et al., 2017) and diminish computational efficiency in scenarios with expert parallelism. Conventional solutions usually rely on the auxiliary loss (Fedus et al., 2021; Lepikhin et al., 2021) to avoid unbalanced load. However, too large an auxiliary loss will impair the model performance (Wang et al., 2024a). To achieve a better trade-off between load balance and model performance, we pioneer an auxiliary-loss-free load balancing strategy (Wang et al., 2024a) to ensure load balance.【deepseek-v3】
    • Auxiliary-Loss-Free Load Balancing.
      • 每个step进行策略调节
      • image-20250501014407504

image-20251003010955657

共享专家和垂类专家

image-20251003010658296

专家容量

image-20251003005711499

SoftMoE

google paper

image-20250703153601233

  • 对于输入的NN个 tokens 通过线性组合(Dispatch)得到SS个 slot,由EE个 Expert 均匀处理SS个 slot 后再映射回(Combine)NN个 tokens,该方案可以看作是某种Merge Tokens的思想。当S<NS<N可显著减少 FLOPS,同时可以通过 Expert 的数目来控制参数量。
    • S == E 时,理解为 Merge Tokens

HardMoE -- PertokensFFN

  • N == S,不再对输入tokens进行dispatch,PertokensFFN
    • 根据语义信息分配token
    • 能保留token间的异构性特点 -- 适用token异构的场景

MoE + Sparsity

参考「MLSys+RecSys」笔记

Long-Context 长上下文

Intro

  • 发展:

    • 早期GPT的上下文只有4K
  • Intro

    • 超大的上下文窗口=超长的短期记忆
    • 128K Token = 124K Input Token + 4096 Output Token

image-20250512021136013

  • 技术路线:
    • Approximation (e.g. Sparse, LoRA)
    • RAG / Vector-DBs (ANN search, LSH)
    • Brute-force compute (tiling, blockwise)

"Train Short, Test Long", Positional Embedding

  • TSTL指的是一种训练和评估大型语言模型(LLM)或其他序列处理模型的方法和期望能力。具体含义如下:

    • Train Short (短序列训练) :在模型训练阶段,主要使用相对较短的文本序列(例如,上下文长度为 512 或 1024 个 token)进行训练。这样做可以:

      • 节省计算资源 :处理短序列需要更少的内存和计算时间,训练速度更快。

      • 利用现有数据 :很多现有的训练数据集可能包含大量中短长度的文本。

    • Test Long (长序列测试/推理) :在模型训练完成后,期望它能够在处理比训练时所见过的序列 更长 的文本时,依然保持良好的性能和稳定性。例如,一个在 1024 token 长度上训练的模型,希望它在处理 2048、4096 甚至更长 token 的输入时,也能理解上下文、生成连贯的文本,并且不会出现性能急剧下降或崩溃的情况。

    • 传统的绝对位置编码(如 Transformer 原始论文中的正弦/余弦编码或学习的绝对位置嵌入)在 TSTL 方面表现不佳。因为它们要么为每个绝对位置学习一个特定的嵌入向量,要么其编码方式在超过训练长度后无法自然外推。当遇到比训练时更长的序列时,模型没有见过这些新位置的编码,导致性能下降。

Alibi

https://arxiv.org/abs/2108.12409

  • 它不直接向词嵌入添加位置信息,而是在计算注意力分数时,给每个 query-key 对添加一个 与它们之间距离成正比的惩罚项(bias) 。
  • 这个惩罚是 相对的 、 局部的 ,并且是 非学习 的(或者说,其斜率是固定的,按注意力头分配)。
  • 因为惩罚只依赖于相对距离,而不是绝对位置编号,所以当序列变长时,这种相对距离的惩罚机制仍然有效。模型自然地倾向于关注更近的 token,这种倾向性不依赖于序列的总长度。因此,Alibi 表现出很好的长度外推能力。

RoPE

苏剑林:Transformer升级之路:2、博采众长的旋转式位置编码

https://arxiv.org/abs/2104.09864

  • Intro
    • 它通过将位置信息编码为 旋转矩阵 ,并应用于 query 和 key 向量。
    • 两个 token 之间的注意力分数依赖于它们向量的点积,而 RoPE 的设计使得这个点积主要取决于它们的 相对位置 (通过旋转角度的差值体现)。
    • 虽然 RoPE 编码的是绝对位置(通过旋转角度),但其核心机制使得相对位置信息得以保留和利用。这种基于旋转的相对位置编码方式,相比于学习绝对位置嵌入,具有更好的外推性,因为它不依赖于为训练长度内的每个绝对位置分配特定编码。

image-20251004225613944

  • 推导
    • q^=f(q,m),k^=f(k,n)\hat q = f(q, m), \hat k = f(k, n)

    • 进一步通过 attention 内积机制实现:

      • f(q,m),f(k,n)=g(q,k,mn)\langle f(q, m), f(k, n) \rangle = g(q, k, m - n)
    • 借助复数域,我们可以将二维下的内积做恒等映射,在复数中, q,k=Re[qk]\langle q, k \rangle = Re[qk^*],所以有映射

      • Re[f(q,m)f(k,n)]=g(q,k,mn)Re[f(q,m)f^*(k, n)] = g(q, k, m-n)
    • 通过对该映射的求解,我们可以得到其复数编码形式与矩阵编码形式,

      • f(q,m)=Rf(q,m)eiΘf(q,m)=qei(Θ(q)+mθ)=qeimθf(q,m) = R_f(q,m)e^{i\Theta_f(q,m)} = ||q||e^{i(\Theta(q)+m\theta)} = qe^{im\theta}
      • f(q,m)=(cosmθsinmθsinmθcosmθ)(q0q1)f(q, m) = \left(\begin{array}{cc} cos m\theta & -sinm\theta \\ sinm\theta & cosm\theta \end{array} \right) \left( \begin{array}{c} q_0 \\ q_1 \end{array}\right)
    • 考虑编码矩阵在多维情况下的稀疏性,所以采用乘法实现,即

      • (q0q1q2q3...qd2qd1)(cosmθ0cosmθ0cosmθ1cosmθ1...cosmθd/21cosmθd/21)+(q1q0q3q2...qd1qd2)(sinmθ0sinmθ0sinmθ1sinmθ1...sinmθd/21sinmθd/21)\left( \begin{array}{c}q_0 \\ q_1 \\ q_2 \\ q_3 \\ ... \\ q_{d-2} \\ q_{d-1} \end{array} \right) * \left( \begin{array}{c} cosm\theta_0 \\ cosm\theta_0 \\ cosm\theta_1 \\ cosm\theta_1 \\ ... \\ cosm\theta_{d/2-1} \\ cosm\theta_{d/2-1} \end{array} \right) + \left( \begin{array}{c}-q_1 \\ q_0 \\ -q_3 \\ q_2 \\ ... \\ -q_{d-1} \\ q_{d-2} \end{array} \right) * \left( \begin{array}{c} sinm\theta_0 \\ sinm\theta_0 \\ sinm\theta_1 \\ sinm\theta_1 \\ ... \\ sinm\theta_{d/2-1} \\ sinm\theta_{d/2-1} \end{array} \right)
RoPE的两种实现:GPT-J style 和 GPT-NeoX style

https://zhuanlan.zhihu.com/p/631363482

其它

LongRoPE, NTK-RoPE, ReRoPE,

也参考 「稀疏注意力」

LWM -- Large World Model with Blockwise Ring-Attn

WORLD MODEL ON MILLION-LENGTH VIDEO AND LANGUAGE WITH BLOCKWISE RINGATTENTION

image-20250512022523757

工程 -> 「LLM-MLSys.md」

半自回归模型

Intro: GPT-4o / Diffusion / MTP

image-20251007010817845

  • MTP、speculative decoding,均属于半自回归的范畴

【《流匹配与扩散模型|6.S184 Flow Matching and Diffusion Models》中英字幕(Claude-3.7-s)-哔哩哔哩】 https://b23.tv/AqSBNNa

和 Diffusion 结合

Google 工作

BlockDiffusion:全局自回归 + 局部非自回归

image-20251007143025845

Bert -- 掩码语言模型

完形填空的训练难度比NTP小

Intro

image-20251003023223287

  • GELU
    • GELUs are used in GPT-3, BERT, and most other Transformers.

image-20241019021744575

Paper

  • Intro
    • BERT: Bidirectional Encoder Representations from Transformers.
    • task类型:sentence-level/paraphrasing/token-level
    • 方法:feature-based and fine-tuning
      • In previous work, both approaches share the same objective function during pre-training, where they use unidirectional language models to learn general language representations.
    • BERT addresses the previously mentioned uni-directional constraints by proposing a new pre-training objective:
      • the "masked language model" (MLM)
      • "next sentence prediction" task

image-20250102001058277

image-20250102001246772

  • 超参:

    • BERTBASE: L=12, H=768, A=12, Total Parameters=110M
    • BERTLARGE: L=24, H=1024, A=16, Total Parameters=340M
    • In all cases we set the feed-forward/filter size to be 4H
    • mask setting:
      • mask 15%,只预测masked词
    • training
      • We train with batch size of 256 sequences (256 sequences * 512 tokens = 128,000 tokens/batch) for 1,000,000 steps, which is approximately 40 epochs over the 3.3 billion word corpus.
      • use Adam with learning rate of 1e-4, β1 = 0.9, β2 = 0.999, L2 weight decay of 0.01,dropout 0.
    • 微调
      • Batch size: 16, 32
      • Learning rate (Adam): 5e-5, 3e-5, 2e-5
      • Number of epochs: 3, 4
  • 模型

    • Emb初始化:We use WordPiece embeddings (Wu et al.,2016) with a 30,000 token vocabulary. We denote split word pieces with ##
    • 设计思想:
      • masked的动机:看到两边,不泄露信息
    • 问题1:训练和微调不一致
      • 方案:8:1:1
      • image-20250102001657033
    • 问题2:每个batch只有15%的token被预测,训练代价大
      • 效果收益更高
    • 任务类型2:next sentence预测,一半对一半
  • 和GPT对比

    • GPT uses a sentence separator ([SEP]) and classifier token ([CLS]) which are only in- troduced at fine-tuning time; BERT learns [SEP], [CLS] and sentence A/B embeddings during pre-training
    • bert训练语料多、batch size大

CLS Token的本质

  • Transformer 具有 field reduce 能力,将 N 个 token reduce 成 M 个 token

model finetune

  • paper
    • squad任务,学一个start和end vector预测start和end位置
    • CoNLL 2003 Named Entity Recognition (NER) dataset
    • swag任务,N选一
      • 学一个V vector
      • image-20250102002146508

image-20250102001936987

  • model finetune是基于BERT预训练模型强大的通用语义能力,使用具体业务场景的训练数据做finetune,从而针对性地修正网络参数,是典型的双阶段方法。(BERT在美团搜索核心排序的探索和实践
  • 在BERT预训练模型结构相对稳定的情况下,算法工程师做文章的是模型的输入和输出。首先需要了解BERT预训练时输入和输出的特点,BERT的输入是词向量、段向量、位置向量的特征融合(embedding相加或拼接),并且有[CLS]开头符和[SEP]结尾符表示句间关系;输出是各个位置的表示向量。finetune的主要方法有双句分类、单句分类、问答QA、单句标注,区别在于输入是单句/双句;需要监督的输出是 开头符表示向量作为分类信息 或 结合分割符截取部分输出做自然语言预测。
  • 搜索中finetune的应用:model finetune应用于query-doc语义匹配任务,即搜索相关性问题和embedding服务。在召回and粗排之后,需要用BERT精排返回一个相关性分数,这一问题和语句分类任务有相似性。搜索finetune的手法有以下特点:
    • 广泛挖掘有收益的finetune素材:有效的包括发布号embedding、文章摘要、作者名,训练手段包括直接输入、预处理。model finetune方法能在标注数据的基础上,利用更多的挖掘数据优化模型。
    • 改造模型输入or输出
      • 模型输入
        • 简单的title+summary+username+query拼接
        • 多域分隔:"考虑到title和summary对于query的相关性是类似的分布,username和query的相关性关联是潜在的。所以给user_name单独设了一个域,用sep分隔"
      • 模型输出
        • 门过滤机制,用某些表示向量的相应分数加权CLS的语句类型输出分
        • 引入UE,直接和CLS输出向量concat
    • 素材的进一步处理,引入无监督学习
      • 在model finetune的有监督训练之前,利用text rank算法处理finetune素材,相当于利用无监督学习提升了挖掘数据 -- 喂入BERT的数据的质量。
      • 截断摘要,实测有效
    • Bert训练任务的设计方式对模型效果影响大
      • 将finetune进一步分为两阶段,把质量较低、挖掘的数据放在第一阶段finetune,质量高的标注数据放在第二阶段finetune,优化finetune的整体效果。
      • 这种递进的训练技巧在BERT中较常见,论文中也有将长度较短的向量放在第一阶段训练的方法。

向量降维

GPT

  • 维特根斯坦:语言是思想的边界

    • NLP是实现AGI的关键
  • 目标:建设NLP领域的"预训练+微调"的训练范式

    • 为什么NLP的研发效率低?
      • 训练速度慢、成本高
      • 任务种类多、繁杂
        • 所有NLP任务都可以转化为语言模型的预测
        • image-20250205180037387
          • Entailment:文本蕴含任务
      • 语料处理难度大
      • 高质量数据稀疏
        • next token prediction任务的泛化性差 --> Scaling Law优化
  • 如何Scaling Law?

    • 简化模型结构:
      • Decoder-Only架构,去除交叉注意力层
        • 6编码6解码 -> 12层解码器,超大参数规模
      • N-gram改变为对全局上下文attention
    • 复杂化模型结构:
      • multi head
      • 增加MLP
      • 多层解码
  • 模型结构:

    • 预训练Loss:取对数,解决seq len增加之后,条件概率的相乘问题
    • 微调Loss:
      • image-20250205191932896

GPT-2

https://cdn.openai.com/better-language-models/language_models_are_unsupervised_multitask_learners.pdf

  • 目标:如果不微调了,能不能有更好的效果?
    • 稀疏自注意力机制
    • 增加batch size到百万,减少通信量
    • 爬取rabbit/wikipedia
  • 思路:足够多的数据,模型能够理解任务意图
    • prompt的前身

image-20241019021839037

GPT-3

  • 目标:根据上下文进行学习

    • image-20250205193747735
  • Decoder

    • 12288维
    • 96层:
      • 12288 -> 128
      • 12288 -> 4*12288
      • Insight:512维存不下96层信息聚合,因此用12288维
N layersDimHeadDim per Head
1.3B24204816128
13B40512040128
175B961228896128

GPT-3.5 (ChatGPT)

Intro

  • 目标:与人类的指令对齐
    • 无法对齐/不安全

image-20250205194626295

  • 对话式大型语言模型:https://openai.com/blog/chatgpt/
    • 自回归语言模型:帮助背下来事件知识

    • 大语言模型:百亿参数以上

      • 不好做finetune,成本高
      • 用prompt作为输入,generated text作为输出
      • 语言知识 + 事件知识,事件知识更需要大模型
    • 未来:AGI(Artificial General Intelligence);教会它使用工具

三个关键技术

  • In-Context Learning 情景学习
    • 在前向中学习
    • 涌现能力:百亿参数规模之后,能力突然提升,改变传统学习范式
    • 大幅降低下游任务开发成本
    • 《Rethinking the Role of Demonstrations: What Makes In-Context Learning Work?》 --> 随机label仍可能提升效果
  • Chain-of-Thought, CoT 思维链
    • 《PAL: Program-aided Language Models》,让语言模型生成能由计算模型执行的描述代码
    • 在大模型中打破scaling law
  • Learning from Natural Instructions 自然指令学习

其它

  • RLHF

    • 见【RLHF】部分
    • 惩罚1:过大的梯度/概率值
    • 惩罚2:灾难性遗忘
  • limitations

    • Correctness: 模型不是全知的,一本正经地胡说八道
    • sensitive to rephrase
    • verbose
    • No asking clarifying questions,而是猜
    • it will sometimes respond to harmful instructions or exhibit biased behavior
  • Iterative deployment

  • Evaluation

    • Holistic Evaluation of Language Models
  • Note

    • 大模型具备了对知识的跨语言能力
    • 科技部部长王志刚表示,ChatGPT有很好的计算方法,同样一种原理,在于做得好不好;就像踢足球,都是盘带、射门,但是要做到像梅西那么好也不容易。
    • 客观题高考515分水平
  • 专访Altman

    • 感想:有几个点值得关注:ai自运行的能力、ai隐藏意图的能力、ai与真实物质世界接口的能力、ai认识到自己的现实处境并差异化处理的能力

      • 当这些能力完全具备,可能AGI确实可以毁灭人类
    • 当他观察模型的隐藏层时,发现它有一个专门的神经元用于分析评论的情感。神经网络以前也做过情感分析,但必须有人告诉它们这样做,而且必须使用根据情感标记的数据对它们进行专门的训练。而这个神经网络已经自行开发出了这种能力。

    • 语言是一种特殊的输入,信息量极为密集

    • "假设我们真的造出了这个人工智能,其他一些人也造出了"。他认为,随之而来的变革将是历史性的。他描述了一个异常乌托邦的愿景,包括重塑钢筋水泥的世界。他说:"使用太阳能发电的机器人可以去开采和提炼它们需要的所有矿物,可以完美地建造东西,不需要人类劳动。"你可以与 17 版 DALL-E 共同设计你想要的家的样子,"Altman说。"每个人都将拥有美丽的家园。在与我的交谈中,以及在巡回演讲期间的舞台上,他说他预见到人类生活的几乎所有其他领域都将得到巨大的改善。音乐将得到提升("艺术家们将拥有更好的工具"),人际关系(人工智能可以帮助我们更好地 "相互对待")和地缘政治也将如此("我们现在非常不擅长找出双赢的妥协方案")。

    • GPT-4学会了"说谎":验证码

      • -> 让GPT-4讲解自己做事情的目的,将不再可靠
      • Sutskever 说,他们可能会在弱小的时候采取一种行动,而在强大的时候采取另一种行动。我们甚至不会意识到,我们创造的东西已经决定性地超越了我们,我们也不知道它打算用自己的超能力做些什么。

GPT-4

  • 亮点:
    • 多模态
    • 大量的RLHF,最安全/可控的模型
    • 在小模型上做消融实验,从而预测大模型实验效果
    • 专家算法投票
  • GPT-4幕后的研发团队大致可分为七个部分:预训练(Pretraining)、长上下文(Long context)、视觉(Vision)、强化学习和对齐(RL & alignment)、评估和分析(Evaluation & analysis)、部署(Deployment)以及其他贡献者(Additional contributions)
  • GPT-4技术报告
    • 32k对应50页的context
  • Scaling Prediction:GPT-4印证了Scaling Law
    • image-20251001200931740

LLAMA 2

{
  "_name_or_path": "meta-llama/Llama-2-7b-hf",
  "architectures": [
    "LlamaForCausalLM"
  ],
  "bos_token_id": 1,
  "eos_token_id": 2,
  "hidden_act": "silu",
  "hidden_size": 4096,
  "initializer_range": 0.02,
  "intermediate_size": 11008,
  "max_position_embeddings": 4096,
  "model_type": "llama",
  "num_attention_heads": 32,
  "num_hidden_layers": 32,
  "num_key_value_heads": 32,
  "pretraining_tp": 1,
  "rms_norm_eps": 1e-05,
  "rope_scaling": null,
  "tie_word_embeddings": false,
  "torch_dtype": "float16",
  "transformers_version": "4.31.0.dev0",
  "use_cache": true,
  "vocab_size": 32000
}

LLAMA 3

https://ai.meta.com/blog/meta-llama-3/

官方代码:https://github.com/meta-llama/llama3

Transformers 库:https://github.com/huggingface/transformers/tree/main/src/transformers/models/llama

官方模型下载(需要申请):https://huggingface.co/meta-llama

TinyLlama:https://huggingface.co/TinyLlama/TinyLlama-1.1B-Chat-v0.4

  • Intro

    • uses RMSNorm [ZS19], SwiGLU [Sha20], rotary embedding [SAL+24], and removes all biases
  • https://hasgeek.com/simrathanspal/the-llama3-guide/sub

  • image-20251004224418837

  • FFN

    • hidden_size 4096
    • intermediate_size 11008
    • hidden_act: SiLU
  • Masked MSA

    • hidden_size 4096
    • num_attention_heads 32
    • num key/value heads 32
  • Decoder

    • LLAMA 7B:num_hidden_layers = 32

DeepSeek

DeepSeek LLM: Scaling Open-Source Language Models with Longtermism

绝大部分是对 llama-2 的复现

  • Intro

    • 7B、67B
    • 2T tokens ~ 67B
    • We further conduct supervised fine-tuning (SFT) and direct preference optimization (DPO) on DeepSeek LLM Base models, resulting in the creation of DeepSeek Chat models
  • data

    • **1.在数据集上改进,**可以不大,但要足够优质;
  • 训练:对scaling law做了比较细致的研究

    • multi-step learning rate scheduler:continue training比较方便,8:1:1
    • 3.1 Scaling Laws for Hyperparameters
      • 关键变量 Compute Budget
      • image-20250503024408398
    • Scaling law的表达,优化chinchella
      • both 6𝑁1 and 6𝑁2 do not account for the computational overhead of attention operation
      • introduced a new model scale representation: non-embedding FLOPs/token M
    • image-20250503025702048
  • Infra

    • Model weights and optimizer states are saved every 5 minutes asynchronously
  • 算法

    • 2.在查询方式上改进(分组查询注意力-Grouped Query Attention,简称 GQA),通过分组查询减少计算复杂度,提高模型性能;
    • 3.深度优先设计(Depth-First Design ,简称DFD),**加高模型层数,**这更类似于人类解题时"一层层"推理的思维方式,使其在数学推理、代码生成等任务中表现更优。
  • evaluation

    • 拒绝刷榜
      • 20 million MC questions
      • exclude MC(multiple choice) data from both the pre-training and fine-tuning stages

DeepSeek-Coder

DeepSeek-Coder: When the Large Language Model Meets Programming -- The Rise of Code Intelligence

  • Continue Pre-Training From General LLM
    • To further enhance the natural language understanding and mathematical reasoning abilities of the DeepSeek-Coder model, we perform additional pre-training from the general language model DeepSeek-LLM-7B Base (DeepSeek-AI, 2024) on 2 trillion tokens, resulting in DeepSeek- Coder-v1.5 7B
    • image-20250503231326154
      • 和仅用代码数据训练的模型对比

DeepSeek-V2

MLA + MoE

image-20250502115344909

image-20250503132537828

  • Each MoE layer consists of 2 shared experts and 160 routed experts, where the intermediate hidden dimension of each expert is 1536. Among the routed experts, 6 experts will be activated for each token.
  • DeepSeek-V2 comprises 236B total parameters, of which 21B are activated for each token.
  • MLA requires only a small amount of KV cache, equal to GQA with only 2.25 groups, but can achieve stronger performance than MHA

Decoupled RoPE

HF Reduce

成本、推理

  • During our practical training on the H800 cluster, for training on each trillion tokens, DeepSeek 67B requires 300.6K GPU hours, while DeepSeek-V2 needs only 172.8K GPU hours, i.e., sparse DeepSeek-V2 can save 42.5% training costs compared with dense DeepSeek 67B.
  • In order to efficiently deploy DeepSeek-V2 for service, we first convert its parameters into the precision of FP8. In addition, we also perform KV cache quantiza- tion (Hooper et al., 2024; Zhao et al., 2023) for DeepSeek-V2 to further compress each element in its KV cache into 6 bits on average
  • On a single node with 8 H800 GPUs, DeepSeek-V2 achieves a generation throughput exceeding 50K tokens per second, which is 5.76 times the maximum generation throughput of DeepSeek 67B. In addition, the prompt input throughput of DeepSeek-V2 exceeds 100K tokens per second.

DeepSeek-Coder-V2

DeepSeek-Coder-V2: Breaking the Barrier of Closed-Source Models in Code Intelligence

  • DeepSeek-Coder-V2 is further pre-trained from an intermediate checkpoint of DeepSeek-V2 with additional 6 trillion tokens. Through this continued pre-training, DeepSeek-Coder-V2 substantially enhances the coding and mathematical reasoning capabilities of DeepSeek-V2, while maintaining comparable performance in general language tasks.

  • Reward Modeling

    • Reward models play crucial roles in the RL training. In terms of mathemat- ical preference data, we obtain them using the ground-truth labels. In terms of code preference data, although the code compiler itself can already provide 0-1 feedback (whether the code pass all test cases or not), some code prompts may have a limited number of test cases, and do not provide full coverage, and hence directly using 0-1 feedback from the compiler may be noisy and sub-optimal. Therefore, we still decide to train a reward model on the data provided by the compiler, and use the reward model to provide signal during RL training,
    • 这个工作仍然没放弃reward model,后面放弃了
      • image-20250503232631642
      • scale up时,reward model有弊端

DeepSeekMath

  • DeepSeekMath 7B, which continues pretraining DeepSeek-Coder-Base-v1.5 7B with 120B math-related tokens sourced from Common Crawl, together with natural language and code data
  • introduce Group Relative Policy Optimization (GRPO), a variant of Proximal Policy Optimization (PPO), that enhances mathematical reasoning abilities while concurrently optimizing the memory usage of PPO
  • 第4章,Reinforcement Learning,值得学习
    • 见「Machine-Learning」-- RL
  • 仍然有reward model
    • We construct the training set of reward models following (Wang et al., 2023b). We train our initial reward model based on the DeepSeekMath-Base 7B with a learning rate of 2e-5. For GRPO, we set the learning rate of the policy model as 1e-6. The KL coefficient is 0.04.

MATH-SHEPHERD: VERIFY AND REINFORCE LLMS STEP-BY-STEP WITHOUT HUMAN ANNOTATIONS

  • follow openai verify step-by-step的过程监督的reward model路线
    • openai PRM800K,过程监督的标注数据
    • deepseek这篇paper:自己构建数据,比如第二步继续往下走的结果,反推第二步是否正确,从而标注第二步
  • present an innovative process-oriented math process reward model called MATH-SHEPHERD, which assigns a reward score to each step of math problem solutions. The training of MATH-SHEPHERD is achieved using automati- cally constructed process-wise supervision data, breaking the bottleneck of heavy reliance on manual annotation in existing work. We explore the effectiveness of MATH-SHEPHERD in two scenarios: 1) Verification: MATH-SHEPHERD is utilized for reranking multiple outputs generated by Large Language Models (LLMs); 2)Reinforcement Learning: MATH-SHEPHERD is employed to reinforce LLMs with step-by-step Proximal Policy Optimization (PPO)

image-20250503235653460

  • image-20250504000341456
    • SC:自己投票,少数服从多数
    • ORM:结果监督而不是过程监督
    • SHEPHERD:过程监督
    • 这个图 本质是早期的 test-time scaling

DeepSeek-Prover

DeepSeek-Prover: Advancing Theorem Proving in LLMs through Large-Scale Synthetic Data

image-20250504004250620

  • LEAN作为formal verifier,很像一个规则
    • 尽管这里是迭代式的自我更新,不是RL

DeepSeek-Prover-V1.5

  • Rewards. When training LLMs via RL, a trained reward model typically provides feedback signals. In contrast, formal theorem proving benefits from the rigorous verification of generated proofs by proof assistants, offering a significant advantage. Specifically, each generated proof receives a reward of 1 if verified as correct, and 0 otherwise.
    • While this binary reward signal is accurate, it is also sparse, especially for theorems that are challenging for the supervised fine-tuned model. To mitigate this sparsity, we select training prompts that are challenging yet achievable for the supervised fine-tuned model, as described above.
    • 针对01 reward信号sparse的问题:把特别难的问题remove掉了
  • Reinforcement Learning Algorithm: GRPO

DeepSeek-V3

DeepSeek-V3 Technical Report

zartbot解读

量子位

腾讯解读,较形象

Intro

image-20251007010333327

  • DeepSeek-V3, a strong Mixture-of-Experts (MoE) language model with 671B total parameters with 37B activated for each token

    • 关键技术
      • Multi-head Latent Attention (MLA)
      • DeepSeekMoE architectures
      • an auxiliary-loss-free strategy for load balancing and sets a multi-token prediction training objective for stronger performance
      • fp8 training
      • DualPipe:overcome the communication bottleneck in cross-node MoE training
      • cross-node all-to-all communication kernels
      • 显存优化
      • MTP
    • 数据量:14T tokens
    • 训练成本:
      • 2048张H800,2.788M H800 GPU hours for its full training
      • 558万刀

    image-20251007010203261

  • 训练框架:

    • On the whole, DeepSeek-V3 applies 16-way Pipeline Parallelism (PP) (Qi et al., 2023a), 64-way Expert Parallelism (EP) (Lepikhin et al., 2021) spanning 8 nodes, and ZeRO-1 Data Parallelism (DP) (Rajbhandari et al., 2020).
      • 没有使用代价很大的TP并行, 这是针对H800被砍了NVLINK带宽的优化
  • 训练流程:

    • pretrain 14T tokens
    • a two-stage context length extension for DeepSeek-V3. In the first stage, the maximum context length is extended to 32K, and in the second stage, it is further extended to 128K.
    • post-training, including Supervised Fine-Tuning (SFT) and Reinforcement Learning (RL)

image-20250501010935207

MLA

Zartbot 解读

todo MLA系列知识点 http://xhslink.com/o/A7ws1PQXhM2

image-20250503133239668

  • The core of MLA is the low-rank joint compression for attention keys and values to reduce Key-Value (KV) cache during inference
    • 从 ht 到 ctKV,进行一次低秩变换
    • dcdhnhd_c \ll d_h n_h

image-20250501011323591

DeepSeekMoE

DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Models (202401)

  • 解决的问题:

    • (1) Knowledge Hybridity: existing MoE practices often employ a limited number of experts (e.g., 8 or 16), and thus tokens assigned to a specific expert will be likely to cover diverse knowledge. Consequently, the designated expert will intend to assemble vastly different types of knowledge in its parameters, which are hard to utilize simultaneously.
    • (2) Knowledge Redundancy: tokens assigned to different experts may require common knowledge. As a result, multiple experts may converge in acquiring shared knowledge in their respective parameters, thereby leading to redundancy in expert parameters. These issues collectively hinder the expert specialization in existing MoE practices, preventing them from reaching the theoretical upper-bound performance of MoE models.
  • --> shared expert、expert变多

    • 结论:With only 40.5% of computations, DeepSeekMoE 16B achieves comparable performance with DeepSeek 7B.
  • shared experts

    • image-20250501014015449
    • image-20250503130340000
  • Auxiliary-Loss-Free Load Balancing

    • 每个step进行策略调节
    • image-20250501014407504
  • Complementary Sequence-Wise Auxiliary Loss.

    • image-20250501021522251
  • Node-Limited Routing.

    • 至多M nodes,每个node选 Kr/M 个专家

MTP

增加了数据的使用效率.

Gloeckle et al. (2024)

  • Different from Gloeckle et al. (2024), which parallelly predicts 𝐷 additional tokens using independent output heads, we sequentially predict additional tokens and keep the complete causal chain at each prediction depth.

image-20250501023813072

  • Our principle of maintaining the causal chain of predictions is similar to that of EAGLE (Li et al., 2024b), but its primary objective is speculative decoding (Leviathan et al., 2023; Xia et al., 2023), whereas we utilize MTP to improve training.

  • the acceptance rate of the second token prediction ranges between 85% and 90%

DualPipe + Efficient communication kernels

  • image-20250501025054344

  • image-20250501025258887

  • customize efficient cross-node all-to-all communication kernels (including dispatching and combining) to conserve the number of SMs dedicated to communication.

    • In detail, we employ the warp specialization technique (Bauer et al., 2014) and partition 20 SMs into 10 communication channels.
    • During the dispatching process, (1) IB sending, (2) IB-to-NVLink forwarding, and (3) NVLink receiving are handled by respective warps. The number of warps allocated to each communication task is dynamically adjusted according to the actual workload across all SMs. Similarly, during the combining process, (1) NVLink sending, (2) NVLink-to-IB forwarding and accumulation, and (3) IB receiving and accumulation are also handled by dynamically adjusted warps.
      • 考虑到带宽差距为3.2倍, 将每个Token最多分发到4个节点减少IB流量
    • In addition, both dispatching and combining kernels overlap with the computation stream, so we also consider their impact on other SM computation kernels. Specifically, we employ customized PTX (Parallel Thread Execution) instructions and auto-tune the communication chunk size, which significantly reduces the use of the L2 cache and the interference to other SMs.
      • 使用cs(cache streaming)策略, 因为这些数据仅在通信时访问一次, 标记在L2 Cache中尽快的被evict.

Fp8-Training、推理部署,参考其它

Pretraining、model结构

  • data

    • Inspired by Ding et al. (2024), we implement the document packing method for data integrity but do not incorporate cross-sample attention masking during training
    • Fill-in-Middle (FIM) strategy does not compromise the next-token prediction capability while enabling the model to accurately predict middle text based on contextual cues
      • image-20250501215109853
    • The tokenizer for DeepSeek-V3 employs Byte-level BPE (Shibata et al., 1999) with an extended vocabulary of 128K tokens.
      • the new pretokenizer introduces tokens that combine punctuations and line breaks. However, this trick may introduce the token boundary bias (Lundberg, 2023) when the model processes multi-line prompts without terminal line breaks, particularly for few-shot evaluation prompts. To address this issue, we randomly split a certain proportion of such combined tokens during training, which exposes the model to a wider array of special cases and mitigates this bias.
  • model

    • We set the number of Transformer layers to 61 and the hidden dimension to 7168. All learnable parameters are randomly initialized with a standard deviation of 0.006.
    • In MLA, we set the number of attention heads 𝑛ℎ to 128 and the per-head dimension 𝑑ℎ to 128. The KV compression dimension 𝑑𝑐 is set to 512, and the query compression dimension 𝑑′𝑐 is set to 1536. For the decoupled queries and key, we set the per-head dimension 𝑑𝑅ℎ to 64. We substitute all FFNs except for the first three layers with MoE layers.
    • Each MoE layer consists of 1 shared expert and 256 routed experts, where the intermediate hidden dimension of each expert is 2048. Among the routed experts, 8 experts will be activated for each token, and each token will be ensured to be sent to at most 4 nodes. The multi-token prediction depth 𝐷 is set to 1, i.e., besides the exact next token, each token will predict one additional token. As DeepSeek-V2, DeepSeek-V3 also employs additional RMSNorm layers after the compressed latent vectors, and multiplies additional scaling factors at the width bottlenecks. Under this configuration, DeepSeek-V3 comprises 671B total parameters, of which 37B are activated for each token.
    • 4.3. Long Context Extension

Evaluation

  • MTP提升效果
  • auxiliary-loss-free balancing strategy提升效果
  • 超过 llama3.1 405B 效果

Post-Training

整体做的比较浅

SFT

  • RL training phase
    • R1生成reasoning data
      • <problem, original response>, <system prompt, problem, R1 response>.
    • Non-Reasoning Data.
      • For non-reasoning data, such as creative writing, role-play, and sim- ple question answering, we utilize DeepSeek-V2.5 to generate responses and enlist human annotators to verify the accuracy and correctness of the data.
  • SFT Settings:We fine-tune DeepSeek-V3-Base for two epochs using the SFT dataset, using the cosine decay learning rate scheduling that starts at 5 × 10-6 and gradually decreases to 1 × 10-6. During training, each single sequence is packed from multiple samples. However, we adopt a sample masking strategy to ensure that these examples remain isolated and mutually invisible.

RL

  • Rule-Based RM.
  • Model-Based RM.
    • The reward model is trained from the DeepSeek-V3 SFT checkpoints. To enhance its reliability, we construct preference data that not only provides the final reward but also includes the chain-of-thought leading to the reward.
    • 仅有开发式问题用奖励模型

其它

  • Distillation from DeepSeek-R1
  • Self-Rewarding

DeepSeek-OCR todo

https://github.com/deepseek-ai/DeepSeek-OCR/blob/main/DeepSeek_OCR_paper.pdf

Embedding Model

https://ezml.io/blog/beyond-clip-the-future-of-multimodal-retrieval-with-visualized-bge-vista-and-magiclens

CLIP -> 参考 MLLM

CoCa

https://research.google/blog/image-text-pre-training-with-contrastive-captioners/

字节跳动 Seed 团队

晚点独家:吴永辉接管字节 Seed 这一年 (2025回顾) 来源:晚点独家丨吴永辉接管字节 Seed 这一年 (公开信源)

  • 背景与目标
    • 2025年初接管,面临追赶压力(投入大、起步晚)。
    • 目标:做到国内第一,与国际领先模型竞争。
  • 管理风格与策略
    • "务实又浪漫":Google 背景(Search -> Brain -> DeepMind),懂技术,能判断方向。
    • 虚拟团队机制
      • Seed Edge:3年考核,攻坚 AGI 长期课题。
      • Focus:打破边界,攻坚下一代模型。
      • Base:负责当前一代模型(工程、数据、测评)。
    • 透明化尝试:推动数据/代码库透明(虽因泄密有所回调),打破"小组制"壁垒。
  • 组织架构 (2025)
    • 负责人:吴永辉(直接汇报给集团管理层)。
    • 多模态/应用:周畅(豆包助手、Seedream 文生图、Seedance 文生视频)。
    • Infra:项亮。
    • AI Lab 并入:李航(负责 AI for Science, Robotics, Responsible AI)。
    • 大模型应用:朱文佳。
  • 成果
    • Seed 迭代 4 版(含豆包 2.0),文生图/视频处于前列。
    • 豆包手机助手成为焦点。

Seed 1.8 通用 Agent 模型

  • 简介:通用 Agent 模型,集搜索、代码与 GUI 能力于一体,原生多模态(图文)输入与界面交互,强调低延迟与高效响应。被评测视为"小号 Gemini",重回国产第一梯队。
  • 核心能力
    • 高效推理:Medium 档位仅需 5K Token 即可达到前代 15K Token 的智力水平,性价比极高;High 档位通过更多思考预算逼近北美头部模型。
    • 多模态融合:坚持统一多模态路线,视觉理解与多模态交互能力优秀。
  • 能力改进
    • 长链推理:相比 1.6 版本 CoT 专注力翻倍,能逐步验证排除分支,适合解决复杂问题(虽然仍偏向暴力穷举,硬智力相比 Gemini 3 Pro 仍有差距)。
    • 信息提取:精度高,但 Token 消耗大(需在 CoT 中完整复述细节),强依赖推理模式。
  • 不足之处
    • 编程能力:虽有起色但工程思维不足,难以作为复杂工程的主模型。
    • 多轮对话:超过 10 轮后思路易发散,无法稳定跟踪任务目标。
    • 空间智力:缺乏训练,处理复杂空间形状问题能力较弱。
  • 评测亮点:GUI Agent 能力在电脑/网页/移动端环境中表现可靠;在 BrowseComp-en 等基准中处于第一梯队。
  • 参考Model Card原文第三方测评

Datasets and Evaluation

Intro

  • 小数据集:快速验证收敛性

20250402-184503

Datasets

  • 100B token:Common Crawl数据集
  • hellaswag, a commonsense sentence completion task
  • wikitext, a next token/byte prediction task, and a few question-answering tasks such as arc, openbookqa, and piqa.
    • For wikitext, perplexity refers to the inverse of how well the model can predict the next word or byte (lower is better), and bits_per_byte refers to how many bits are needed to predict the next byte (lower is also better here). For all other tasks, acc_norm refers to the accuracy normalized by the byte-length of the target string.
  • Dolma:3T token https://huggingface.co/datasets/allenai/dolma

Evaluation

模型能力

image-20250616165815319

BPB

  • BPB (Bits Per Byte) 与模型的交叉熵损失(Cross-Entropy Loss)直接相关,这源于它们在信息论和模型评估中的基本含义。以下是详细解释:
  1. 交叉熵损失的含义 :

    • 在语言模型中,交叉熵损失衡量的是模型预测的下一个字节(或 token)的概率分布与真实下一个字节的概率分布之间的"距离"或差异。
    • 具体来说,对于一个给定的上下文,模型会输出一个概率分布 Q ,表示它预测下一个字节是词汇表中每个可能字节的概率。真实的下一个字节对应一个"one-hot"分布 P (真实字节的概率为 1,其他为 0)。
    • 交叉熵损失计算的是 -sum(P(byte) * log(Q(byte))) 。由于 P 是 one-hot 的,这简化为 -log(Q(actual_next_byte)) ,即模型赋予真实发生的下一个字节的概率的负对数。
    • 关键点 :**这个 -log(Q(actual_next_byte)) 值,从信息论的角度来看,可以解释为:根据模型 Q 的预测,编码(表示)实际发生的那个字节 actual_next_byte 所需要的信息量(比特数,如果对数以 2 为底)。**损失越低,意味着模型赋予真实字节的概率越高,编码它所需的信息量就越少。
  2. BPB 的含义 :

    • BPB 定义为:模型平均需要多少比特(bit)来编码输入文本中的每一个字节(byte)
  3. 两者之间的联系 :

    • 模型的平均交叉熵损失(Average Cross-Entropy Loss)计算的是在整个数据集上,模型编码每个真实字节所需的 平均信息量 。
    • 如果交叉熵损失是以 2 为底的对数( log2 )计算的,那么这个平均损失值 直接就是 BPB。因为 log2 计算的结果单位就是比特(bit)。
    • 在深度学习实践中,交叉熵损失通常使用自然对数( ln 或 log_e )计算,得到的单位是奈特(nats)。由于 log2(x) = ln(x) / ln(2) ,因此: BPB = AverageCrossEntropyLoss_nats / ln(2)
    • 所以,无论使用哪个底数的对数,平均交叉熵损失都直接(或通过一个常数 ln(2) 转换)对应于 BPB。

个性化能力

  • PersonaMem -- 记忆个性化评测

In-context Learning

https://ai.stanford.edu/blog/understanding-incontext/

SFT (Supervised Finetuning)、对齐

Intro

  • 数据工程是 SFT 的核心
  • 如何提升数据数量和质量是数据工程的核心
    • 从 0 到 60 分:习得输出的模板,激发预训练中习得的知识
    • 从 60 到 100 分:提升推理、生成和知识性能力,进一步对齐人类的期望
  • 在未来,如何保证数据的多样性可扩展性是 SFT 成功的关键
  • 如何做finetune
    • 流程和DL的训练差不多
      • 数据预处理阶段,会加载tokenizer,将文本转token ids
    • 基座模型选型
    • 全参数finetune和小参数量finetune
      • 小参数量finetune
        • Adapters
        • Prompt-tuning v1/v2
        • LoRA
  • Finetune的需求:
    • 场合:私有部署+开源模型能力不足
    • 数据量需求:
      • OpenAI: 1.3w条SFT prompt
      • embedding:至少10w条数据,相似性和同义性

Literature Review

  • finetuning分类

    • full:Training Language Models to Follow Instructions with Human Feedback

      • aligned with human preferences with instruction-tuning
    • 高效的:LoRA: Low-Rank Adaptation of Large Language Models

  • Pre-trained LLMs can be adapted to domain tasks with further fine-tuning

    • 《Large language models encode clinical knowledge》
  • fine-tuned LLMs fail to learn from examples

    • DAIL-SQL

LoRA

image-20231026212212239

https://github.com/huggingface/peft

Prefix Tuning & P-Tuning

特性Prefix TuningP-Tuning v1P-Tuning v2
论文Li & Liang (ACL 2021)Liu et al. (GPT-3 authors)Liu et al. (ACL 2022)
作用位置每一层 Transformer (K/V)仅输入层 (Embedding)每一层 Transformer (Prompts)
机制结构Deep Modification (深层)Shallow Modification (浅层)Deep Modification (深层)
参数化方式MLP 重参数化 (仅训练时)LSTM/MLP Encoder (关键)无重参数化 (直接优化)
适用任务NLG (生成)NLU (分类/LAMA)全能 (NLU+NLG)
微调参数量0.1% ~ 3%<0.1%0.1% ~ 3%

Prefix Tuning

  • 核心思想: 在 Transformer 的每一层 Multi-Head Attention 的 Key/Value 之前拼接可训练的 prefix 向量。
    • 公式:hi=Transformer(z,h<i)h_i = \text{Transformer}(z, h_{<i}),其中 zz 是 prefix。
    • Deep Modification: 每一层都加,直接影响每一层的特征表示。
  • 重参数化 (Reparameterization):
    • 直接优化 PP 参数不稳定。作者引入 MLP (Matrix-Vector) 来生成 PPP=MLP(θ)P = \text{MLP}(\theta')
    • 训练后丢弃 MLP,只存 PP
  • 优缺点:
    • NLG 任务(如 GPT-2 生成)效果极好。
    • NLU 任务上效果一般。

P-Tuning v1

  • 核心思想: 将自然语言 Prompt 替换为可学习的连续 Embedding,仅插入在输入层
  • Prompt Encoder:
    • 发现直接训练 Embedding 难收敛(离散性与关联性问题)。
    • 使用 LSTM/MLP 对 pseudo embeddings 进行建模,使得 Prompt token 之间具有依赖关系。
  • 局限: 由于只在输入层修改,对于深层模型(如 100+ 层的 LLM),Prompt 的影响力随着层数加深而迅速衰减,导致在复杂的 NLU 任务(如序列标注)上表现不佳。

P-Tuning v2

  • 核心思想: Deep Prompt Tuning。吸取 Prefix Tuning 的"深层"思想,解决了 v1 的痛点。
  • 机制:
    • Deep Prompting: 像 Prefix Tuning 一样,在 Transformer 的每一层都插入可训练的 Prompt 向量(通常视作在序列前加 token)。
    • 移除重参数化: V2 发现,在 Deep 设定下,不需要 v1 中的 LSTM/MLP Encoder,直接优化 Prompt 参数即可收敛。
  • 优势:
    • 全能: NLU 和 NLG 任务都接近全量微调。
    • 解决难题: 在序列标注(Sequence Labeling)等硬 NLU 任务上表现大幅提升。
    • 适用性: 在小模型(如 300M)到大模型上都有效(v1 在小模型上效果差)。

P-tuning vs Prompt Engineering 本质区别

  • Prompt Engineering (Hard Prompt):

    • 离散空间搜索: 在人类可读的自然语言空间中寻找最佳 Token 组合(如 "Translate this to French:")。
    • 不涉及训练: 模型参数完全冻结,只改变输入文本。
    • 局限: 容易陷入次优解,对措辞极其敏感(微小的词汇变化导致巨大输出差异)。
  • P-tuning (Soft Prompt):

    • 连续空间搜索: 在高维连续的向量空间中学习最佳 Embedding 参数。
    • 参数微调: 这一串 Embedding 不需要对应具体的自然语言单词,通过反向传播更新这些 Prompt 参数(模型主体参数通常冻结)。
    • 优势: 突破了自然语言的离散限制,能找到人类语言无法描述但对模型最优的"触发器"。

Mid Training

为什么 Mid-training 在 Agent 时代变得如此关键?

刘鹏飞团队:Mid-training是Agent时代胜负手,全球首个全面开源 链接:https://mp.weixin.qq.com/s/jUIR_5XUfZH1nMkjemqx_g

  • 在推理(Reasoning)时代,Mid-training(中期训练)还只是"锦上添花";到了Agent时代,Mid-training从"锦上添花"变成了"不可或缺"。
  • Agent时代的能力需求,已经超出了Post-training的能力边界:
    • 在传统的单轮对话或推理任务中,模型需要的是单一能力的纵深——把推理做到极致,把对话做得流畅。
    • 但在Agent场景下,模型面对的是多能力的动态编排:一个完整的软件开发任务,可能需要串联代码定位、bug诊断、测试生成、工具调用、上下文管理等10+种能力,每个环节都可能触发不同的子任务分支。
  • Post-training的逻辑是:在预训练模型已有的能力边界内,通过人类(或AI)反馈、强化学习等方式优化表现。它可以让模型更好地"对齐"人类(AI)偏好,可以纠正特定错误模式,但它无法凭空创造模型不曾见过的能力项。
  • 更致命的是,长链条任务的容错率极低:在一个包含20步的开发流程中,每步成功率95%,最终成功率只有35%。任何一个环节的能力缺失,都可能导致整个任务失败。
  • Post-training擅长修补"已知问题",但面对"未知的能力缺口"——比如从未学过的"工具调用后的异常处理",再多的Post-training也无济于事。
  • 而这些能力基建,必须在Mid-training阶段通过大规模、高复杂度的长链条数据来构建。

现有训练数据的根本性缺陷:分布偏差

  • 目前的开源代码模型,大多是用GitHub上的代码数据训练出来的。但这里有一个巨大的分布偏差(Distribution Mismatch)。
  • GitHub上的代码(Pull Request),通常是静态的"快照":它告诉你"结果是什么",却没告诉你"过程是怎么来的"。
    • 它没记录开发者为了改这行代码,翻看了哪三个相关文件(Context);
    • 它没记录开发者第一次改错了,看了什么报错信息(Traceback),才修正成最终版本;
    • 它没记录开发者如何在测试失败后,迭代修正代码直到通过。
  • 用静态的结果去训练动态的Agent,就像是给想学开车的人看了一万张"汽车停在车库里"的照片,却从来不带他上路。

Agent-Native Data:还原"思考"与"体感"

为了填补这一鸿沟,构建了两类核心数据,重塑了模型对软件工程的认知:

Contextually-native Trajectories(上下文原生轨迹,686亿token):还原"思考流"

  • 不仅仅是代码补全。团队从1000万+真实GitHub PR中,重构了代码变更背后的完整程序化过程。
  • 关键创新:Bundle Everything Together(捆绑一切上下文)
    • 模拟定位:模型需要学习开发者是如何在几百个文件中找到由于依赖关系需要修改的那一个;
    • 模拟推理:还原开发者"阅读 Issue -> 分析现有代码 -> 构思修改方案"的完整心路历程。
    • 这让模型不再是"盲写",而是学会了先看后写(Navigation before Editing)。

Environmentally-native Trajectories(环境原生轨迹,31亿token):还原"真实体感"

  • 这是最残酷但也最有效的一环。团队在2万个真实Docker环境中部署Agent,让它"真刀真枪"地干活。
  • 关键创新:Real Execution Loop(真实执行循环)
    • 真实交互:调用 Linter,运行 Unit Tests,拿真实的 Build System 交互。
    • 真实反馈:模型看到的不是人工构造的完美数据,而是真实的 Runtime Error。它必须学会根据报错去自我修正。
    • 这赋予了模型"自我反思"的本能。

两类数据的协同效应:1+1>2

  • 团队的实验揭示了一个关键发现:两类数据形成互补,缺一不可。
  • 这说明:
    • PR数据提供"知识和广度":跨语言、跨框架的软件工程模式,让模型见识足够多的"什么样的问题对应什么样的解决思路";
    • 环境轨迹提供"深度和真实性":动态交互和反馈循环,让模型习得"如何在不确定环境中迭代求解"。
  • 就像学开车:PR数据是"科目一理论学习",环境轨迹是"科目三实际上路"。两者结合,才能培养出真正的Agent。

实验结果:以小博大,效率倍增

基于 Qwen 2.5 Base 模型进行了验证,结果令人振奋:配方(Recipe)的重要性,甚至超越了单纯的数据堆砌。

01 效率的胜利:数据减半,性能反超

  • 相比于此前开源界最先进的 Kimi-Dev 训练配方,我们仅使用了其 50% 的数据量(73.1B vs ~150B tokens),就在两种后训练方式均下实现了性能反超。这说明,Agent-Native 数据的"含金量"远高于普通的合成数据。

02 跨代际的打击

  • 在 SWE-Bench Verified(公认的真实软件工程评测)上,我们的 daVinci-Dev-72B 和 daVinci-Dev-32B 模型分别达到了 58.5% 和 56.1% 的解决率。
  • 更值得注意的是,在 Agentic Scaffold(智能体框架)下,该模型的表现超越了目前基于 Qwen 2.5 Coder 甚至 Qwen 3 的开源训练方案。
  • 这揭示了一个深刻的道理:与其盲目追求更新的基座,不如用更符合 Agent 本能的数据去重塑现有的模型。

03 能力的泛化:Agent思维的溢出效应

  • 虽然团队专注于软件工程,但Agent的决策和规划能力展现出了惊人的迁移性(科学推理、科学计算),这说明,在Agent化过程中习得的"定位-推理-行动-反思"范式,本身就是一种通用的问题解决能力,可以迁移到其他需要多步推理的领域。
  • 模型在Mid-training中学到的不只是"如何写代码",更是"如何分解问题、寻找信息、验证假设、迭代优化"——这是Agent的底层操作系统。

Pre/Mid/RL 在推理LM中的作用(Interplay)

  • 背景:现代RL显著提升LM推理,但是否真正超越预训练能力存在争议;预训练语料不透明使因果归因困难。
  • 方法:构建可控框架,用合成推理任务(显式原子操作、可解析步骤轨迹)与系统化分布操控,沿两条轴线评估:外推泛化(更复杂组合)与上下文泛化(跨表层语境)。
  • 术语:OOD(Out-of-Distribution)指超出模型训练/假设数据分布的样本或任务;与 ID(In-Distribution)相对。
  • 术语:长尾语境与"暴露"比例
    • 长尾语境:在预训练语料中出现频率低但对目标任务关键的语境(如法律文本、低资源语言、特定格式/风格、长链路推理样式、程序化符号操作)。
    • 暴露比例 pexposure[0,1]p_{exposure}\in[0,1]:指预训练数据中属于目标长尾语境的样本占比(按样本数、token数或任务实例数度量)。
    • 近零暴露:pexposure0%p_{exposure}\approx0\%(如 <0.1%),模型几乎未见该语境;RL 难以迁移。
    • 稀疏暴露:pexposure1%p_{exposure}\geq1\% 但远小于主分布,模型具备最低限度的语境先验;RL 可稳健迁移。
    • 最小但充分暴露:使模型在预训练阶段获得必要的语境先验与表示支撑,达到 RL 可转移的临界点(经验阈值约在 ≥1%)。
  • 结论1(能力增益的前提):只有当预训练留有"能力余量"且RL样本定位于模型能力边界(略难但可达)时,RL才产生真实增益;若任务已被预训练覆盖或过于OOD,增益消失(最高可达 +42% pass@128)。
  • 结论2(上下文泛化的最小暴露):上下文泛化需要预训练阶段对长尾语境的最小但充分暴露;近零暴露时RL失败,稀疏暴露(≥1%)即能让RL稳健迁移(最高 +60% pass@128)。
  • 结论3(固定算力下的中期训练价值):在固定计算预算下,引入中期训练(mid-training,亦称CPT)作为分布桥接显著提升OOD推理;"mid + RL"较"仅RL"在 OOD-hard 上提升约 +10.8%。
  • 结论4(过程级奖励):过程级奖励(奖励推理步骤正确性)减少 reward hacking,提升推理一致性与保真度。
  • 实践建议:
    • 难度校准:采样位于能力边界的RL数据,避免过易/过难。
    • 语境暴露:保证预训练中对关键语境≥1%的稀疏覆盖。
    • 预算分配:边界内任务更多预算给mid-training;边界外难题更多预算给RL
    • 奖励设计:采用过程级奖励以减少投机取巧。

Instruction tuning

Intro

  • 典型例子ChatGPT

  • 提升模型泛化能力

image-20251002023902315

Literature Review

  • Zero-shot training of retrievers.

    • 克服没见过的任务的query难点
      • 无监督:leveraging another model to automatically generate training data (Wang et al., 2022a).[TRwI]
      • 生成label(template-based):Dai et al. (2022) use task-specific templates and few-shot samples to automatically generate in-domain training queries given randomly sampled documents from the target corpus using FLAN (Wei et al., 2022a)..[TRwI]
  • Instruction Tuning

    • Weiet al., 2022a; Sanh et al., 2022; Ouyang et al., 2022; Min et al., 2022; Wang et al., 2022b; Mishra et al., 2022; Chung et al., 2022 .[TRwI]

    • 缺少指令tuning的retrieval[TRwI]

      • 缺少标注数据集
      • llm生成海量emb的成本高
      • Retrieval with descriptions的路线:效果一般
    • dataset scale提升instruction的泛化能力

      • Recent work (Wang et al., 2022b; Chung et al., 2022) show that scaling up the number of the training datasets improves LLMs' ability to adapt to new task via instructions. We open-source our instruc- tion data and call for community efforts to collect more retrieval tasks and human-written instructions as in instruction-following for LMs (Wang et al., 2022b; Bach et al., 2022), to investigate whether further increasing the number of the datasets (e.g., more than 100 datasets) improves zero-shot and cross-task retrieval. [TRwI]

组合泛化 + 大参数量:1+1>2

image-20251002024429291

Alpaca & Vicuna: 数量 v.s. 质量

image-20251002025358987

image-20251002025703039

image-20231025213448602

LIMA: Less Is More For Alignment

image-20251002030109269

image-20251002030131953

image-20251002030145476

UltraChat: Scalable Diversity的数据集

image-20251002030324827

image-20251002030504238

image-20251002030443463

Orca:从传统NLP任务中挖掘指令数据

image-20251002030655684

Alignment

https://github.com/tatsu-lab/stanford_alpaca

指令微调是什么? - superpeng的回答 - 知乎 https://www.zhihu.com/question/603488576/answer/3178990801

  • 指令微调是一种特定的微调方式,在不同的论文中以不同的方式引入。我们在一个新的语言建模任务上对模型进行微调,其中的示例具有额外的结构,嵌入到模型提示中。
    • 先无监督训练,再用有监督的"指令-回答"预料
    • 指令调整模型接收一对输入和输出,描述引导模型的任务。
  • 核心思路:解决"回答问题"与"接话"的差异
  • Note:
    • 数据获取昂贵(RLHF人工打分的成本比人工写故事要低)
    • 对开放性问题效果不好(write a story about ...)

FoodGPT: A Large Language Model in Food Testing Domain with Incremental Pre-training and Knowledge Graph Prompt

  • Incremental Pre-training 增量预训练

    • 图像和扫描文档
      • 存储大量领域标准文档信息,使用 OCR 技术处理。因文档可能超模型训练序列长度,按章节拆分,为防描述冲突,给数据章节添加前缀(通过 UIE 模型提取文档名,启发式生成方法构建前缀)。同时用 BERT 和 GPT - 2 计算文本章节中句子的困惑度,排除高困惑度句子。
    • 结构化知识
      • 存在于私有结构化数据库,由人工输入的表格组成。创建 Datav1 和 Datav2 两个版本用于增量预训练。Datav1 去除机密隐私信息后用字典构建数据,以 "测试项目" 为键,对应多个具体测试项目的表格(markdown 格式)为值;Datav2 采用新方法序列化,去除机密隐私信息后合并部分无单独意义的字段,输入 ChatGPT 按规则随机生成文本。
    • 其他类型数据
      • 包括食品检测字典、中国食品检测教程和研究论文、食品情感数据、食品安全相关法律、食品安全相关考题等,选择 Chinese - LLaMA2 - 13B 为基础模型,用 LoRA 方法进行增量预训练。
  • Instruction Fine-tuning

    • 数据集构建

      • 通过两种方式构建指令微调数据集。一是从食品论坛选取相关主题,抓取大量问答对,优先选择发帖频率高的用户以确保高质量答案;二是与食品检测领域专家合作设计 100 个高质量种子指令,用 evol - instruct 方法扩展和多样化。
    • 训练过程

      • 用 LoRA 方法对 Chinese - LLaMA2 - 13B 的指令进行微调。

RLHF -- 基于人类反馈的强化学习

  • 参考「Reinforcement-Learning」

预训练小模型、模型融合

TODO 前阿里、字节大模型带头人杨红霞创业:大模型预训练,不是少数顶尖玩家的算力竞赛|智能涌现独家

MLLM(Multimodal LLM)

TODO InfiniTensor 2024冬 多模态分享: https://www.bilibili.com/video/BV1r3fJYVEre

Intro

  • Modal: 图片、视频、音频、文本

  • MLLM = LLM + 接收、推理多模态信息的能力

    • 听雨声,判断路面情况,今天是否适合出门

    • 概念:单模态、多模态、跨模态、多模态语言大模型

    • 单模态

      • image-20241124014848392
      • LVM
    • 跨模态:

      • 音频->视觉:数字人
        • 蚂蚁Echomimic:实时渲染口播
        • 快手:LivePortrait
          • 非人、卡通,都能驱动
        • SadTalker paper/code
        • 浙大、字节 Real3d-portrait
        • ani-portrait
        • facebook research:audio2photoreal
      • 文本->音频:
        • TTS、音色克隆、少样本:GPT-SoVITS

          • 情感色彩、语调,一般
        • ChatTTS

          • 有情感色彩
        • SUNO:音乐生成

        • 开源工具

          • Meta:audiodraft
          • stable-audio-open-1.0
    • 多模态模型

      • image-20241207210031737

Literature Review

  • Vision Transformers [Beyond the CLS Token: Image Reranking using Pretrained Vision Transformers]
    • Vision Transformers (ViT) [9], directly applied transformer architectures from NLP to image classification.
    • To improve the training efficiency of ViT, DeiT [28] introduced token-based distillation with Convolutional Neural Networks (CNNs) as the teacher.
    • combine CNNs and ViT
      • PVT [30] introduced the pyramid structure into ViT, which generates multi-scale feature for dense prediction tasks.
      • CvT [33] leveraged convolutional patch embedding and convolutional attention projection to combine the best aspects of both CNNs and transformers.
      • The Swin Transformer [18] introduced a shifted window scheme to limit self-attention within windows while allowing interaction between windows.

多模态大模型历史发展

ViT模型,图像表示的token化

ViT

image-20241207210214783

image-20241207210250921

[ViT-MAE] Vision Transformer based on Masked Autoencoding (Kaiming He)
  • In the input image, 75% patches are randomly masked; the encoder module of ViT only takes unmasked patches as input, and produces an embedding. This embedding is then concatenated with learnable masked image patch encoding.
  • img
Swin Transformer: Hierarchical Vision Transformer using Shifted Windows
  • key differences between language and vision data is of the variation in scale between image features and language tokens.

    • visual feature的尺度更细; nlp token的尺度固定
  • img

  • SWIN is a hierarchical transformer which addresses this problem of scale variation by computing transformer representation with shifted windows. The idea is to further divide usual image patches of input image to even smaller patches. These smaller non overlapping patches are then presented to attention layers.

  • The output from these attention layers are then concatenated in pairs to combine attention output the two higher level patches, this concatenated output is presented to next set of attention modules.

  • This hierarchical propagation through attention layers, allows transformer to pay attention to smaller scale features and deal with variation in scales for image data.

    • brings greater efficiency by lim- iting self-attention computation to non-overlapping local windows while also allowing for cross-window connection.
    • 解决transformer复杂度O(N^2)的问题

image-20241218022713658

image-20241218023502807

image-20241218023820301

  • Efficient batch computation for shifted configuration
    • Cyclic shift
  • 其它
    • relative position bias
    • Table 5 研究了 Real speed of different self-attention computation meth- ods and implementations on a V100 GPU
SWIN v.s ViT

基于transformer的图像-文本联合建模

  • VisualBert

image-20241207210505919

  • BEit
    • img

大规模图文Token对齐模型 CLIP

What is CLIP?

CLIP (Contrastive Language-Image Pre-training; Radford et al. 2021) jointly trains a text encoder and an image feature extractor over the pretraining task that predicts which caption goes with which image.

CLIP, developed by OpenAI, is a model designed to understand and relate images and text through contrastive learning. It learns to match images with their corresponding text descriptions and to differentiate these pairs from mismatches, enabling it to perform various tasks, from image classification to zero-shot learning.

How Does CLIP Work?

  • Contrastive Learning: CLIP is trained on a vast dataset of image-text pairs, learning to create a shared embedding space where both images and texts are represented as vectors. The model maximizes the similarity of correct image-text pairs and minimizes it for incorrect pairs.
  • Joint Embedding Space: CLIP's ability to create a joint embedding space for images and text allows it to generalize across different tasks and domains.

Limitations of CLIP

  • Fine-Grained Visual Understanding: CLIP struggles with fine-grained visual details due to its broad learning approach. It can miss subtle distinctions within images that are critical for certain tasks.
  • Imprecise Multimodal Alignment: The alignment between text and images can be imprecise, especially when dealing with complex or nuanced relationships.
  • Retrieval Performance Variability: CLIP's performance can vary depending on the specificity of the query and the image, sometimes leading to suboptimal results.

image-20241207210538634

Given a batch of NN (image, text) pairs, CLIP computes the dense cosine similarity matrix between all N×NN \times N possible (image, text) candidates within this batch. The text and image encoders are jointly trained to maximize the similarity between NN correct pairs of (image, text) associations while minimizing the similarity for N(N1)N(N-1) incorrect pairs via a symmetric cross entropy loss over the dense matrix.

See the numpy-like pseudo code for CLIP in:

# image_encoder - ResNet or Vision Transformer
# text_encoder  - CBOW or Text Transformer
# I[n, h, w, c] - minibatch of aligned images
# T[n, l]       - minibatch of aligned texts
# W_i[d_i, d_e] - learned proj of image to embed
# W_t[d_t, d_e] - learned proj of text to embed
# t             - learned temperature parameter

# extract feature representations of each modality
I_f = image_encoder(I) #[n, d_i]
T_f = text_encoder(T)  #[n, d_t]

# joint multimodal embedding [n, d_e]
I_e = l2_normalize(np.dot(I_f, W_i), axis=1)
T_e = l2_normalize(np.dot(T_f, W_t), axis=1)

# scaled pairwise cosine similarities [n, n]
logits = np.dot(I_e, T_e.T) * np.exp(t)

# symmetric loss function
labels = np.arange(n)
loss_i = cross_entropy_loss(logits, labels, axis=0)
loss_t = cross_entropy_loss(logits, labels, axis=1)
loss = (loss_i + loss_t) / 2

Compared to other methods above for learning good visual representation, what makes CLIP really special is "the appreciation of using natural language as a training signal". It does demand access to supervised dataset in which we know which text matches which image. It is trained on 400 million (text, image) pairs, collected from the Internet. The query list contains all the words occurring at least 100 times in the English version of Wikipedia. Interestingly, they found that Transformer-based language models are 3x slower than a bag-of-words (BoW) text encoder at zero-shot ImageNet classification. Using contrastive objective instead of trying to predict the exact words associated with images (i.e. a method commonly adopted by image caption prediction tasks) can further improve the data efficiency another 4x.

image-20251220033339399

CLIP produces good visual representation that can non-trivially transfer to many CV benchmark datasets, achieving results competitive with supervised baseline. Among tested transfer tasks, CLIP struggles with very fine-grained classification, as well as abstract or systematic tasks such as counting the number of objects. The transfer performance of CLIP models is smoothly correlated with the amount of model compute.

image-20241207210618154

BLIP-2: 表征学习 + 文生图、Q-Former、Image Captioning

https://www.nematilab.info/bmijc/assets/081823_paper.pdf

  • 核心:通过轻量级 Q-Former 连接冻结的图像编码器与冻结的 LLM,实现高效图文联合建模
    • 一阶段:学习图像和文本域的表征,对比学习+判别式+生成式loss
    • 二阶段:文本生成预测任务
    • 一二阶段都训练 Q-Former 的参数
  • 支持 instructed zero-shot image-to-text generation

BLIP-2 采用「双冻结 + 桥接」设计,避免端到端训练千亿级参数模型的高成本:

  • 冻结组件
    • 图像编码器(如 ViT-L/14):提取图像视觉特征
    • 大语言模型(如 FlanT5, LLaMA):负责文本生成
  • 桥接组件:Q-Former(Query Transformer)-- 唯一可训练的轻量级模块

Q-Former 是一个小型 Transformer,包含:

  • 可学习的查询向量(Query Tokens)
    • 数量固定(如 32 个),作为图像与文本交互的「中介」
    • 通过自注意力与交叉注意力同时建模视觉特征和文本序列
  • 双编码器结构
    • 视觉编码器:接收图像编码器输出的 patch 特征,与查询向量交互,生成「视觉感知查询特征」
    • 文本编码器:接收文本序列,与查询向量交互,生成「文本感知查询特征」
image-20251114160143557 *Q-Former 与冻结图像编码器的交互:查询向量通过交叉注意力提取图像特征*

训练阶段

image-20251114175550796

Q-Former 通过两阶段训练实现图文对齐,每个阶段使用不同的损失函数组合:

  1. 阶段一:视觉-语言表示学习 (Vision-Language Representation Learning) 此阶段的目标是训练 Q-Former,使其可学习的查询向量(Queries)能够从冻结的图像编码器中提取出对文本最有效、最相关的视觉表征。为此,模型联合优化三个共享参数但采用不同注意力掩码机制的目标函数,总损失为 L=Litc+Litm+LitgL = L_{itc} + L_{itm} + L_{itg}

    • 图文对比学习 (Image-Text Contrastive, ITC):

      • 目标:对齐图像和文本的全局表征,最大化其互信息。
      • 机制:通过对比学习,让正样本(匹配的图文对)的相似度远高于负样本(不匹配的图文对)。具体地,计算每个查询向量输出与文本[CLS]表征的相似度,并选取最高分作为最终的图文相似度。
      • 掩码策略:采用单向自注意力 (Unimodal Self-Attention),查询向量和文本序列之间不可见,以学习各自模态的独立表示。
      • 损失函数Litc=12i=1N(H(yii2t,pii2t)+H(yit2i,pit2i))L_{itc} = \frac{1}{2} \sum_{i=1}^{N} \left( H(y^{i2t}_i, p^{i2t}_i) + H(y^{t2i}_i, p^{t2i}_i) \right),其中 pp 是 softmax 归一化的相似度,yy 是 one-hot 标签,HH 为交叉熵。由于图像编码器被冻结,可以采用 in-batch negatives 提高效率。
    • 图文匹配 (Image-Text Matching, ITM):

      • 目标:学习图像与文本间的细粒度对应关系。
      • 机制:这是一个二分类任务,判断图文对是正样本(匹配)还是负样本(不匹配)。
      • 掩码策略:采用双向自注意力 (Bi-directional Self-Attention),所有查询向量和文本 Token 之间可以相互关注,使得查询向量的输出能够捕获跨模态信息。每个查询向量的输出都会经过一个分类头,最终将所有头的 logits 平均作为匹配分数。
      • 损失函数Litm=E(I,T)[ylogσ(p(I,T))+(1y)log(1σ(p(I,T)))]L_{itm} = - \mathbb{E}_{(I,T)} \left[ y \log \sigma(p(I,T)) + (1-y) \log(1-\sigma(p(I,T))) \right],其中 p(I,T)p(I,T) 是预测的匹配分数,yy 是真值标签。此任务采用难负例挖掘 (hard negative mining) 来提升判别能力。
    • 图像引导的文本生成 (Image-Grounded Text Generation, ITG):

      • 目标:强制查询向量包含生成对应文本所需的全部视觉细节。
      • 机制:由于 Q-Former 的架构限制了图像编码器和文本 Token 的直接交互,生成文本所需的信息必须先由查询向量提取,再通过自注意力层传递给文本 Token。
      • 掩码策略:采用多模态因果自注意力 (Multimodal Causal Self-Attention)。查询向量之间可以相互关注,但不能看到文本;每个文本 Token 可以关注所有查询向量及其前面的文本 Token。
      • 损失函数Litg=E(I,T)k=1TlogP(TkT<k,I)L_{itg} = - \mathbb{E}_{(I,T)} \sum_{k=1}^{|T|} \log P(T_k | T_{<k}, I),在给定图像 II 的条件下,自回归地预测文本 TT
  2. 阶段二:视觉到语言生成学习 (Vision-to-Language Generative Learning) 此阶段的目标是训练 Q-Former 作为 LLM 的"视觉提示"生成器。

    • 语言模型损失 (Language Modeling Loss): 将第一阶段训练好的 Q-Former 输出的查询向量 ZZ 作为软提示(soft prompt),与外部输入的文本提示 TpromptT_{prompt} 一起送入冻结的 LLM,以生成答案 TanswerT_{answer}Lgen=E(I,Tprompt,Tanswer)k=1TanswerlogPLLM(Tanswer,kTanswer,<k,Z,Tprompt)L_{gen} = - \mathbb{E}_{(I, T_{prompt}, T_{answer})} \sum_{k=1}^{|T_{answer}|} \log P_{LLM}(T_{answer,k} | T_{answer,<k}, Z, T_{prompt}) 损失函数是标准的交叉熵损失,仅在答案 TanswerT_{answer} 的 token 上计算,以此教会 LLM 理解 Q-Former 编码的视觉信息。

image-20251114160444328

  • 核心优势
    • 参数效率:仅训练 Q-Former(约 1% 参数),避免微调千亿级 LLM/图像编码器
    • 模态桥接:查询向量同时建模视觉与文本特征,解决模态鸿沟问题
    • 泛化能力:支持零样本任务(如零样本图像描述、视觉问答)

多模态大语言模型

  • GPT-4v

    • 遵循文字指令

    • 理解视觉指向和参考

    • 支持视觉+文本联合提示

    • few-shot

    • 视觉认知能力强

    • 时序视觉信号理解

  • Gemini:原生多模态大模型

image-20241207211915709

  • Gemini 3

    • Reference: Gemini3预训练负责人访谈
    • 模型架构
      • Transformer MoE:核心架构未变,强调模块化复用与组合。
      • 非单一突破:性能跃迁是大中小多重因素(旋钮微调)叠加及大规模团队协作改进的累积。
    • 系统工程
      • 研究即工程:构建围绕神经网络的复杂系统,边界模糊化。
      • Pre+Post:核心优势源于"更好的预训练"叠加"更好的后训练"(Post-training,如RLHF/RLAIF)。
    • 趋势观点
      • 数据范式:从"无限数据"进入**"数据受限"**阶段,合成数据(Synthetic Data)成为Scaling关键燃料。
      • Scaling Law:短期内(1年+)持续有效,Benchmark难度提升但模型智能仍在增长。
  • GPT-4o

    • GPT 4o本质上是要探索不同模态相互融合的大一统模型应该怎么做的问题,对于提升大模型的智力水平估计帮助不大
  • Kimi K2.5

    • Reference: https://www.zhihu.com/question/1999487395494588876/answer/2000362433332660070?share_code=HGYNzglkEV8h&utm_psn=2000559952910099841
    • 统一的视觉-视频 Encoder 设计(N+1 思路)
      • 架构选择:SigLIP NaViT(内部名 MoonViT),设计思路是做真正的 unified encoder
      • 核心原则:image 和 video 必须共享同一个 encoder,参数完全共享
      • 为什么需要 3D encoder:有些 motion 和细微的时序变化(快速手势、瞬间表情转换),如果只抽 2D 帧,后面靠 LLM 去"脑补"时序信息,到模型里已经看不清了
      • 为什么要共享 encoder
        • 如果 image 和 video 用两套独立 encoder,要么伤害 image 本身能力,要么 video 学不到真东西
        • embedding 会打架:图片里能认出"特朗普",到视频里可能就"变傻"了
        • image pretrain 学到的所有知识(物体识别、空间关系、世界知识)在 video 场景下都能完整保留
        • video 只需要额外学习时序和动态这部分独特的能力
    • Post-training 的多模态融合——同样的 N+1 思路
      • 前提:base model 首先 vision 能力很强,同时 K2 Thinking(纯文本模型)的 thinking 和 tooluse 能力也很强
      • 核心思路:vision 能不能直接继承 text 的 thinking 和 tooluse pattern?
      • 实现方案:用极小的、没有 vision 输入的冷启动数据(zero-vision cold-start)去激活这种能力,让 vision 直接对齐到 text 的 thinking 和 tooluse pattern 上
      • 好处
        • 模型不仅能看懂图,还能保持灵活的 reasoning toolcall 能力
        • 不需要训练 <VideoCut> <Crop> <Rotate> 这种特别的 tool,直接用 IPython tool 就行
        • 能激活神奇的走迷宫能力
    • Vision RL
      • 做了非常 holistic 的 vision-centric RL,分涨了很多
      • 连 agentic 的 trajectory 都变好了
      • 随着 grounding training set accuracy 的提升,模型在 IPython 里天然就 crop 得越来越准
    • 最神奇的发现:Vision RL 能让模型"长知识"
      • 本来以为 vision RL 更多是教模型提取信息(怎么用 image 里的信息避免模型不看图瞎说)
      • 但 RL 完一测发现,文本知识涨了!
      • 原因:vision 用的是 text 教出来的 thinking pattern,没"分脑",就"泛化"了
      • 后来整个 post-train vision-text joint RL 的环节通常都是一半甚至更多的 vision 数据配比

Data Prepare

image-20241207212813240

  • Trick
    • image放在prompt结尾,比较少受文本信息干扰

image-20241207212906560

  • prompt

image-20241207215105555

Training - Llava

Intro

  • 模型:

    • ViT的倒数第二层除cls token外的image token
  • 细节:

    • 容易过拟合,--num_train_epochs=1,一般是从头训练

image-20241207212730097

image-20241207213002988

算法迭代

  • 改进Visual Encoder
    • image-20241207215512977
    • image-20241207215556328
    • image-20241207215612284
    • image-20241207225347798
  • 改进Projection Layer
    • lora思想、改进文本能力
    • image-20241207230013814

视频、语音输入

image-20241207230602139

原生MLLM

  • Next-GPT训练
    • 阶段一:更新input projection layer
    • 阶段二:decoder段输出结果与指令对齐,只更新output projection layer
    • 阶段三:

image-20241207231036385

image-20241207231155505

image-20241207230626127

image-20241207230702605

image-20241207230732346

image-20241207230840157

image-20241207230853889

image-20241207230909253

image-20241207230919593

image-20241207230938271

image-20241207230956810

image-20241207231244953

开源项目

image-20241207230532748

image-20241207230211854

image-20241207230235619

Evaluation

应用于 切图、物体匹配

Talking to DINO: Bridging Self-Supervised Vision Backbones with Language for Open-Vocabulary Segmentation

技术关键点和结论:

  • 利用ViT patch embedding,作为图像的局部特征
  • 通过learned mapping,将图像的局部特征和clip category embedding对齐,做实体分割
  • 核心思路:通过learned mapping,对vit patch embedding和clip category embedding对齐,做实体分割

image-20241213203708145

  • 算法:
    • warp text embedding
    • Dinov2:
      • N个attention map(patch维度)
      • N个weighted visual embedding
      • N个相似度分数
    • 对比学习:最相似的weighted visual embedding <-> text embedding
    • Identifying Background Regions

image-20241214121142744

image-20241214121543467

[todo] OmniGlue: Generalizable Feature Matching with Foundation Model Guidance

  • 技术关键点和结论(仅略读):
    • Google的CV领域SOTA paper,基于更强的Foundation Model做优化
    • 针对图像Feature Matching的场景,DIML技术,用optimal transport做排序

Applications

  • 工业

  • 医疗

  • 视觉内容认知与编辑

  • 具身智能

  • 新一代人机交互

  • 多模态Agent

    • CogAgent
      • 围绕GUI的能力强化:解析和目标定位能力
  • Llava衍生的应用

    • 图表问答生成:ChartLlama-code

image-20241207213052536

视频算法

Intro

image-20241207231303500

  • 一锤降维!解密OpenAI超级视频模型Sora技术报告,虚拟世界涌现了
    • 扩展视频生成模型的规模,是构建模拟物理世界通用模拟器的非常有希望的方向

    • patch

      • 从宏观角度来看,研究者首先将视频压缩到一个低维潜空间中,随后把这种表征分解为时空patch,这样就实现了从视频到patch的转换。
      • 在推理时,可以通过在一个合适大小的网格中适当排列随机初始化的patch,从而控制生成视频的大小。
    • 训练技巧

      • 直接在视频原始比例上训练
      • 研究者采用了DALL·E 3中的重新标注技术,应用在了视频上。
        • 首先,研究者训练了一个能生成详细描述的标注模型,然后用它为训练集中的所有视频,生成文本说明。
        • 他们发现,使用详细的视频说明进行训练,不仅能提高文本的准确性,还能提升视频的整体质量。
        • 类似于DALL·E 3,研究者也使用了GPT,把用户的简短提示转化为详细的说明,然后这些说明会被输入到视频模型中。
    • 生成的视频特点:

      • 多种输入形式、多视频间过渡、人和物的特征

视频抽关键帧

Literature Review

  • 方法:
    • uniform sampling based,
    • clustering based,
      • VSUMM [4], SGC [5], GMC [6] used k-means, minimum spanning tree, and graph modularity
      • 缺点是忽略了temporal sequences
    • comparison based,
      • VSUKFE [7] and DiffHist [8]
      • 根据阈值对比
    • shot based approaches
      • drawing only one frame from each shot is insufficient to fully describe videos' visual contents;
      • using traditional features for boundary detection might be inaccurate for shot segmentations.

Large Model based Sequential Keyframe Extraction for Video Summarization

  • 切片(shot):TransNetV2
  • 帧理解:CLIP
  • 每个shot内的frame聚类
    • 迭代出k_max个聚类中心
      • kmax=Nk_{max}=\sqrt{N}
    • 最大化SC(silhouette coefficient),合并聚类中心
      • 聚类中心合并到2个,选择SC最大的一个聚类 (类比于筛掉一半聚类对应的帧,并选择聚类效果更好的一个中心)
    • Redundancy Elimination
      • 先基于color histogram去除solid-color or uninformative frames
      • 再基于color histogram迭代去除相似帧

image-20250109174239161

image-20250109181815631

  • benchmark构建
    • 人工打分,取局部极值点作为关键帧

An effective Key Frame Extraction technique based on Feature Fusion and Fuzzy-C means clustering with Artificial Hummingbird

  • https://www.nature.com/articles/s41598-024-75923-y
  • 和 LMSKE 的差异(二者均为一个hybrid方案):
    • 先利用 【颜色通道相关性、直方图差异、互信息、惯性矩】筛选关键帧再做聚类
      • LMSKE:shot切分 -> 聚类(利用多模态Embedding) -> 筛选(颜色通道)
      • 该paper:筛选(多种特征) -> 聚类(利用HSV)
    • 聚类算法的改进:Artificial Hummingbird、Fuzzy C-means Clustering
  • 优劣势分析:相比LMSKE,实时性更好、视频图片语义信息的利用更少

视频理解

  • TS2NET:soft/hard attn,做max pooling进行筛选token

图像和视频生成

视频生成模型基础:从时空 latent 到迭代生成

当前主流视频生成模型常把原始视频压缩到 latent space,再在低维时空表示上生成。也存在自回归 video token 和混合路线;下文聚焦目前常见的 latent diffusion / flow 架构:

文本 / 图像 / 音频条件
        ↓ encoder
真实视频 → Video VAE Encoder → 时空 latent / spacetime patches
                                  ↓ Diffusion Transformer / Flow Transformer
随机噪声 → 多次预测更新方向 → clean latent
                                  ↓ Video VAE Decoder
                               最终视频

Video VAE 与 spacetime patch

  • Video VAE 同时压缩空间分辨率和时间长度,避免直接在所有 RGB 像素上训练与采样;
  • latent 再切成时空 patch,形成 Transformer 可处理的 token;
  • token 不只表示“图像的一个小块”,还带有时间位置,因此位置编码要同时表达帧、宽和高;
  • DiT 里的 Transformer 通常是 denoiser / velocity predictor,不等于 LLM 那种逐 token 自回归解码器。

这种表示可以统一不同分辨率、宽高比和时长的视频,但序列长度仍随空间和时间共同增长,计算量远高于单张图像。

Diffusion 与 Flow Matching

Diffusion 训练把 clean latent 扰动到不同噪声强度,再学习反向恢复;Flow Matching 则在噪声分布与数据分布之间选定概率路径,采样路径上的中间状态并回归对应向量场。常见预测参数化包括:

目标直觉
noise当前 latent 中加入了多少噪声
clean sample原始 clean latent 应是什么
score当前概率密度上升最快的方向
velocity从噪声分布流向数据分布的瞬时方向

Flow Matching 直接回归连续概率路径上的向量场:

dztdt=vθ(zt,t,c)\frac{d z_t}{dt}=v_\theta(z_t,t,c)

其中 z_t 是时间 t 的 noisy latent,c 是文本、参考图或其它条件。推理从高斯噪声出发,用数值求解器多次调用模型,逐步走到可解码的视频 latent。Diffusion 与 Flow Matching 的数学参数化不同,但工程心智模型相近:模型学习每一步往哪里走,sampler 决定怎样沿这条路径走。

一次 denoiser / velocity model 调用称为一次 NFE(Number of Function Evaluations)。NFE 越高,通常越容易获得稳定细节,但延迟和成本也越高;少步蒸馏的核心就是让 student 用更少 NFE 跨过近似相同的生成路径。

视频比图像多出的难点

维度典型问题
单帧视觉质量纹理、清晰度、构图、审美
时间一致性闪烁、形变、物体突然出现或消失
运动合理性速度、接触、遮挡、人体动作与物理约束
主体一致性人脸、服装和物体在跨帧、跨镜头后保持身份
镜头与叙事运镜、景别切换、多镜头因果和场景连续性
条件控制prompt、首尾帧、reference、轨迹和音频是否被正确遵循

因此视频模型的能力不只来自 backbone。完整链路通常还包括视频筛选与重标注、渐进式分辨率/时长训练、SFT、视频 reward model / 偏好优化、rejection sampling、少步蒸馏,以及 VAE、attention 和 GPU kernel 优化。只看最终模型结构,无法解释全部质量差距。

参考:DDPMFlow Matching for Generative ModelingSora:Video generation models as world simulatorsSeedance 1.0 技术报告

rejection sampling

本质是 **"生成一批样本,筛选后保留优质样本" 的后处理方法 **。

论文中用于类条件(ImageNet)和文本条件(CC-3M)图像生成:

  • 背景:自回归模型(如 RQ-Transformer)生成样本时,受 "预测误差累积" 影响,部分样本会出现细节模糊、类别偏离(如类条件生成中 "猫" 生成成 "狗")或文本 - 图像不对齐(如文本 "雪山汉堡" 生成成 "普通汉堡")的问题;
  • 操作逻辑:先让模型生成远多于目标数量的样本(如需要 50K 最终样本,先生成 200K),再用一个 "质量评判标准" 对这些样本打分,拒绝低分劣质样本,只保留高分优质样本,最终得到符合目标数量的高质量样本。
  • 论文中的 "评判标准":类条件生成用预训练的 ResNet-101 分类器(判断生成样本是否符合目标类别),文本条件生成用 CLIP 模型(判断生成图像与文本的对齐度)。

评估使用 acceptance rate

视频生成模型蒸馏:白盒轨迹压缩与黑盒成片模仿

视频生成模型并非不能蒸馏。需要区分两类目标:

目标Teacher 暴露的信息本质
少步生成 / 推理加速权重、score / velocity、噪声状态和去噪轨迹真正的 diffusion / flow distillation
从闭源 API 复刻能力prompt 与最终成片合成数据模仿,通常不是过程蒸馏

扩散或 flow model 的核心知识不是某一张最终成片,而是任意噪声状态和时间点上的更新方向:

vT(xt,t,c)v_T(x_t,t,c)

白盒蒸馏可以让 student 在相同条件、噪声和中间状态上逼近 teacher 的 score、velocity 或完整轨迹,用更少的 NFE(Number of Function Evaluations)跨过原来的多步去噪路径。这里常被压缩的是采样步数,student 不一定同时拥有更少参数。Consistency distillation、trajectory distillation 和 distribution matching 都属于这条路线;工业视频模型也已用多阶段蒸馏显著降低推理成本。

黑盒 API 只返回终点:

x0pT(xc)x_0 \sim p_T(x\mid c)

它没有暴露“模型怎样从噪声走到该视频”。用这些成片继续训练当然可行,但更接近构造 synthetic dataset,主要受四个问题限制:

  1. 多解性:同一 prompt 有大量合理视频,逐像素不同不代表语义或质量错误。
  2. 高维时空分布:画风和常见构图较容易模仿,运动、遮挡、身份一致性、镜头连续性和物理规律需要远多于单帧的覆盖。
  3. 模式覆盖不足:一次 API 返回只是条件分布中的一个样本;看不到其它生成 mode、失败候选和 reward model 淘汰的负例。
  4. 过程监督缺失:拿不到 score、teacher logits、相同 noise 下的轨迹与中间 latent,监督从密集过程信号退化成稀疏终局样本。

与 LLM 蒸馏的差异主要来自生成目标,而不是“一个用了 Transformer、另一个没有”:

自回归 LLM扩散 / Flow 视频模型
学习离散的 `p_T(next tokenprefix, prompt)`
白盒时可逐 token 匹配 logits白盒时可在多个时间点匹配 score / velocity
最终文本仍可拆成 token 级 SFT 监督最终成片只暴露整条轨迹的终点
黑盒蒸馏同样损失 logits,但文本是高度压缩的语义表示视频包含高维空间、时间和运动自由度,输出模仿的样本复杂度更高

因此,结构差异会放大难度,但不是不可蒸馏的理论边界。决定能否逼近 teacher 的关键变量是:

teacher access × 轨迹监督密度 × 条件分布覆盖 × student 容量 × 自有数据和 reward 系统

一句话:有白盒 teacher 时,视频模型可以做轨迹级少步蒸馏;只有成片时,只能从终点反推分布,能追近风格和高频能力,却很难低成本复制运动模型、控制能力和偏好边界。

参考:Seedance 1.0 技术报告Accelerating Video Diffusion Models via Distribution MatchingConsistency Trajectory ModelsSequence-Level Knowledge DistillationOn-Policy Distillation of Language Models

音视频联合生成

概述
  • 视频生成领域早期以生成无声视频为主,后来 veo3、sora2、Seedance 2.0 等模型实现了音画同步,但大多闭源
  • MOVA:复旦大学发布的开源音画同出模型,是少数接近商用的开源选择,完全开放推理和训练微调代码

参考:

传统方案 vs 联合生成方案

传统方案:流水线拼接

  • 4个模型串联,推理开销高
  • 误差累积,每一步的错误传递给下一步
  • 时序对齐困难,音画同步靠后期硬凑
  • 各模型独立训练,风格难统一

联合生成方案:端到端一体化

  • 视频和音频在生成过程中互相感知,信息融合
  • 一个模型同时输出画面和声音,无需后期拼接
MOVA 技术架构
image-20260216031935350

核心架构:双塔 DiT + 条件桥接,包含:

  • 多语种 prompt 解析模块 UMT5
  • Video DiT Block:负责视频建模
  • Audio DiT Block:用于音频建模
  • Bridge CrossAttn:桥接交叉注意力机制
  • Video/Audio 编解码器

核心创新:Dual-Tower DiT

  • 传统多模态模型通常单向:先生成视频,再根据视频生成音频,后生成的模态只能被动适配
  • MOVA 的 Dual-Tower Conditional Bridge 实现双向条件控制
    # 伪代码示意
    video_tokens = video_self_attention(video_tokens)
    audio_tokens = audio_self_attention(audio_tokens)
    # 双向交叉注意力
    video_tokens = video_cross_attention(video_tokens, key=audio_tokens, value=audio_tokens)
    audio_tokens = audio_cross_attention(audio_tokens, key=video_tokens, value=video_tokens)
    

时序对齐:RoPE 频率同步

  • 音频和视频时间尺度不同:
    • 视频:24 fps,每秒 24 帧
    • 音频:44100 Hz / 2048 ≈ 21.5 步/秒
  • 通过 build_aligned_freqs() 构建对齐的 RoPE 频率,将两者映射到统一时间坐标系,确保时序对齐

实时交互生成:从 next-token 到 next-reaction prediction

来源:智能涌现:AI圈最神秘的一家公司,终于公开了它的“豪赌”,2026-07-21;一手资料:Vivix-W1 Technical ReportVivix-A1 Technical Report

离线视频生成把 prompt、图像或音频作为初始条件,一次性生成完整 clip;实时交互生成则把用户的新动作持续写入生成过程,让当前状态在不重启的情况下改变后续音视频流。核心建模对象从“下一帧长什么样”扩展为“给定历史状态和当前交互,世界接下来如何反应”。

创始人在访谈中把训练单元称为 next reaction prediction,包含三个部分:

  1. 理解:过去发生了什么,包括场景、人物、声音和交互历史;
  2. 交互:用户做了什么,包括文本、语音、触控、镜头与角色运动等原生信号;
  3. 生成:交互发生后,下一段画面、声音、动作和叙事如何变化。

游戏 replay、仿真日志等数据可以记录初始状态和完整操作序列,天然形成 state + action -> next state 的自监督样本。它比“视频转文字、再用文字控制视频”保留了更多时间、动作和环境信息。但 Vivix 尚未公开具体训练目标、数据混合比例和消融实验,next reaction prediction 目前更像数据范式与任务定义,而不是可复现算法。

Streaming-native 模型

  • W1:约 30B active parameters 的中等规模版本。模型按细粒度音视频 segment 持续生成,新交互可在生成过程中进入 context;同时建模音画联合生成、多镜头叙事、多模态 reference 和长时序一致性。Vivix-Turbo 再用低步数蒸馏、长序列连续性优化、误差累积抑制和 reference consistency 降低实时推理成本。
  • A1:面向角色交互,将传统 LLM -> TTS -> avatar 串联管线改成原生多模态生成 loop,并用 Director Agent 做高层规划。短段之间的 local continuity prior 维持局部连贯,global sequence prior 维持长程身份和场景一致性;联合蒸馏把生成压到 2 steps,并同时约束 latent 与 raw data space。

实时生成的系统目标不只是画质。W1 报告估算视频交互的有效 token throughput 比文本高约 \1030^{3}-$10^{4}$$$ 倍,因此需要同时优化:

interactive utility=f(latency,cost,quality,consumption value)\text{interactive utility}=f(\text{latency},\text{cost},\text{quality},\text{consumption value})

访谈给出的产品口径是 TTFF 约 300–600 ms、端到端 TTFR 约 1 s,单位成本以 YouTube 每小时约 0.2 美元的 CDN/带宽成本为目标。官方技术报告只确认 streaming generation、second-level latency 和低步数蒸馏,没有公开可复现的成本—延迟—质量联合 benchmark;这些数字应视为厂商自述的阶段性工程指标。

这里的“世界模型”是功能主义定义:模型根据可观测的音视频状态和人类动作生成下一状态,服务互动叙事、游戏、直播或数字人。它不等于学习完整物理规律,也不能仅凭 demo 证明开放世界中的因果理解。当前 W1 的限制仍包括专业画质、复杂运动和密集场景;A1 仍会出现动作不一致、交互漂移与物理 grounding 错误。

VLA (Vision Language Action) and Robot Foundation Model(具身智能)

> 逐篇解析机器人基座模型和VLA经典论文--"人就是最智能的VLA" -- 张小珺商业访谈录 > 年末 AI 回顾:从模型到应用,从技术到商战,拽住洪流中的意义之线 > 整理时间:2026-02-17

> 清华大学交叉信息研究院助理教授、星动纪元创始人陈建宇,PreA轮

关键词:VLA、Robot Foundation Model、投资与上市潮、具身智能三要素、落地应用

  • Intro
    • LLM的发展触发具身智能的创业潮
    • AlphaGo,MCTS是在连续空间内,机器人也在连续空间内决策,启发了机器人

投资与上市潮:具身智能的中国优势

根据中国信通院《具身智能发展报告(2025)》,截至 25 年底,中国具身智能和机器人领域的年度融资总额已高达 735 亿元。对比之下,几家头部大模型公司(含智谱、MiniMax IPO 融资)的同期融资总额约为 182 亿元。

火热也体现在估值上。在美国,具身公司的估值远低于大模型公司,如最贵的 Figure 估值 390 亿美元,是 OpenAI 的 1/20。而在中国,两者并驾齐驱:银河通用在 25 年底估值已达到 30 亿美元;而即将于 26 年上半年 IPO 的宇树科技,市场对其市值预期甚至直指 500 亿乃至千亿元人民币。

同时,源源不断的新具身团队仍在涌现,25 年新成立的公司就有:从华为、百度自动驾驶部门走出的陈亦伦、李震宇创立了它石智航;旷视联创唐文斌等人创立了 "原力灵机";理想前自动驾驶技术研发负责人贾鹏等人创立的至简动力;华为诺亚方舟实验室前首席研究员李银川创立的诺因知行;月之暗面前强化学负责人宋鸿涌创立的 Android 16;以及星海图联创许华哲,也正在筹划新一次创业。

为什么具身智能在中国格外火热?

除了技术变化的驱动,还有三个原因:

  1. 政策与制造业红利:具身智能有硬件本体,是地方政府招商引资的 "舒适区",能落地看得见的产线。全国已建成及在建的 "具身智能训练场" 已接近 30 家,这种 "遥操作采集数据" 的场景本身就带动了具身智能机器人的初期收入和应用落地。

  2. 供应链比较优势:中国成熟的供应链能显著降低本体成本。例如,宇树科技推出的 10 万元级人形机器人,已成为全球实验室的主流开发工具。

  3. 更明确的退出路径:中国二级市场对制造业更友好。除了宇树,智元、银河通用、星海图等公司据传均计划在 26 年冲击 IPO,他们多选择港股。即使是像智元机器人收购上纬新材股权这种尚未完成实质 "借壳" 的动作,也能让后者的市值从 30 亿暴涨至 500 亿以上。

一批具身公司计划上市,港股宏观行情可能发生波动,以及很多公司还在亏损--这几个因素碰到一起--这场具身上市潮会如何发展?会成为 26 年非常值得关注的一个行业悬念。

具身智能三要素:数据、模型与本体

具身智能进展可被观察的 3 个核心指标是:数据、模型和硬件本体。

其中,数据和模型,是和智能能力直接相关的。行业的共识是,数据是当前的最重要课题,更准确说,是如何规模化且相对低成本地获取大量、有效的数据。

数据获取方式

在怎么获取数据上,现在是八仙过海、各显神通。主要的方式有以下几种:

  1. 通过遥操作来获取真机数据:这个方式需要造很多机器人,投入比较大;
  2. 在仿真环境里获得数据,再迁移到真机上:即 Sim-to-real;
  3. 从视频里获得数据
  4. UMI(universal manipulation interface):主要是通过让人在做任务时,戴上手套等可穿戴设备,来采集手部位姿、力控等数据;
  5. 让机器人自己做任务,失败后自己调整:即通过 self-play 获得数据。

目前流派纷呈:有侧重 "真机遥操作" 的,有侧重 "仿真迁移(Sim-to-Real)" 的(如银河通用、Hillbot),也有利用 "视频学习" 或 "穿戴设备(UMI)" 采集数据的。尽管对于 "仿真数据是否是大坑" 仍有分歧,但组合多种数据源已成主流。

模型技术路线

在模型上,当前行业相对主流的技术路线有 VLA、端到端,还有常被提及的世界模型,它们不是平行概念。

  1. VLA 模型(Vision-Language-Action):目前的主流路径,即通过多模态 VLM 训练出直接输出机器人动作的神经网络。
  2. 端到端:试图用一个深度神经网络解决从感知到规控的全过程。
  3. 世界模型:现在大家主要探索的方向是 "生成式的世界模型"--从世界的这一个状态,预测和生成世界的下一个状态。
    • 如果以 2D 视觉信息表达,是可以无限延续的视频生成模型;所以当 OpenAI 发布 Sora 时,便有人认为这是世界模型的雏形。
    • 若以 3D 视觉信息表达,便是 Google 在 25 年发布的 Genie 3。它能生成一个可供探索的 3D 空间,并配合 Google 的另一个 AI 项目 SIMA 2,让用户创建的 Agent 在其中自由移动。
    • 而真正被期待的 "完整的世界模型",是能实现与环境和物体的直接交互--比如当你戳破一只气球或摘下一朵花时,系统能符合物理规律地预测并生成交互后的下一个状态。

硬件本体

硬件本体则是一个多学科交织的复杂系统工程。

非常推荐《晚点》25 年 10 月发布的一篇报道:《特斯拉人形机器人再延期,因为双手只能用六星期》(李梓楠),深入还原了第三代 Optimus 设计延期背后的供应链细节,解释了为何当时 Optimus 的灵巧手寿命极短、故障率高,且由于设计原因无法局部修理,一旦损坏只能整体更换。此外,整个机身还面临着手臂与腿部关节的稳定性、减重以及续航等重重挑战。一位被 Optimus 屡次拖延的供应商吐槽:"老马(Elon Musk)的信誉分,现在恐怕连充电宝都借不出来了!"

具身智能三要素与当前短板(GR-3 文章)

  • 三要素: 本体 (Embodiment)模型 (Model)数据 (Data),三者需互相依赖、共同迭代。
  • 当前短板: 模型和数据。文章犀利地指出,当前机器人难以胜任各类任务的最大短板是"脑子笨"。一个核心论据是:人类远程遥操作机器人的能力远超任何自主模型。
  • 对本体设计的启示: 机器人本体的设计不应"孤立"进行,而要与当前模型和数据的能力相匹配。过于超前的本体在"笨脑子"的驱动下也无法发挥价值。

典型案例

来源:写在GR-3之后 - 知乎

  • 模型亮点:

    • 通用抓取 (Generalizable Pick-and-Place): 能理解抽象语言指令。
    • 长序列任务 (Long-Horizon Table Bussing): 能纯端到端完成长程任务,无需传统规划。
    • 灵巧操作 (Dexterous Cloth Manipulation): 能双臂协同操作柔性物体(如叠衣服)。
  • 行业洞察:

    • 辩证看待发展:
      • 短期不能高估: 现有具身智能模型的"智能"水平仅相当于1-2岁婴儿,VLA/VTLA模型成熟度远低于LLM/VLM。机器人的运动能力(locomotion)的巨大进步不等于机器人智能的突破。
      • 长期不能低估: 相比5年前基于规则(rule-based)的方法,AI+机器人的能力已实现质的飞跃,能完成过去无法想象的复杂任务。
    • 具身智能三要素与当前短板:
      • 三要素: 本体 (Embodiment)模型 (Model)数据 (Data),三者需互相依赖、共同迭代。
      • 当前短板: 模型和数据。文章犀利地指出,当前机器人难以胜任各类任务的最大短板是"脑子笨"。一个核心论据是:人类远程遥操作机器人的能力远超任何自主模型。
      • 对本体设计的启示: 机器人本体的设计不应"孤立"进行,而要与当前模型和数据的能力相匹配。过于超前的本体在"笨脑子"的驱动下也无法发挥价值。

DexGraspVLA:共享自主与VLA的灵巧抓取范式

  • 概述:字节跳动 Seed 团队提出的层级式 VLA(Vision-Language-Action)框架,结合共享自主采集与扩展触觉,解决数据采集瓶颈与泛化难题。
  • 共享自主采集(Human-in-the-loop)
    • 人控机械臂 6-DoF,AI 控灵巧手 12-DoF;采集效率≈110条/小时,开发周期≈1天。
    • 失败纠正闭环:失败时人类介入,系统自动积累"难例",持续提升策略稳健性。
    • image-20251218202650427
  • 两阶段训练(AI 副驾驶)
    • 盲抓策略:仅依赖触觉,实现从纸杯到 1.2kg 重物的鲁棒抓取。
    • 多模态融合:视觉+语言+双触觉特征,形成域不变表示,降低分布偏移后用模仿学习稳定训练。
  • 特征增强与臂手解耦
    • 臂手动作解耦并融合,整体成功率由 88%→95%;遮挡场景从 19%→58%。
  • 触觉的重要性(XHAND)
    • 无触觉 21%→单触觉 70%→双触觉 90%,触觉显著提升抓取与接触状态估计。
  • 泛化与跨硬件迁移
    • 杂乱场景成功率≈95.5%,新物体零样本泛化≈85.6%。
    • 跨硬件迁移:零样本到 RY-H2 仍≈81% 成功率。
  • 能力特性
    • 长时序自然语言指令执行、对抗性物体与人为扰动鲁棒、失败恢复。
  • 关键设计洞察
    • 用基础模型迭代将多样语言与视觉输入映射到域不变表示,再用模仿学习高效泛化。
  • 引用

落地应用:从实验室走向 "陪伴"

26 年初,智元机器人宣布实现了 5000 台的销量;而宇树则称其纯人形机器人 25 年的实际出货量超 5500 台(不含四足和轮式),本体量产下线已超 6500 台。

目前的落地方向主有 5 个:

1. 研发

目前的交付大头依然是卖给具身智能训练场、高校实验室及研究机构。研发需求也是真的需求。只是在 25 年这波训练场建设热潮中(中国已建成和在建的数采工厂已有 30 座),需要甄别那些名为 "智能训练"、实为 "工业园地产" 的项目。

2. 表演与展示

25 年 7 月,中国移动下达了总额 1.24 亿元的人形机器人采购大单,其中智元拿到了 7800 万,宇树拿到了约 4600 万。这些机器人除了用于机房巡检,很大一部分功能就是展厅接待和营销宣传。

市场上也已出现专门租赁宇树机器人的公司。据报道,靠商演收取的租金,最快两周到一个月就能收回本体成本。不过很多视频里机器人的酷炫动作,其实仍由真人近距离遥控完成,而非机器人自主完成。

3. 商业与家庭服务:最热门却难啃

研发和表演需求都有阶段性,也有比较明显的规模上限,长期大家想实现的,还是让机器人进入工厂、商店甚至家庭里,自己干活。

虽然 Sunday Robotics 或 1X 的原型机在视频里表现惊人--比如叠衣服、拿高脚杯--但在真实的餐厅、酒店或家庭里,我们依然很难见到它们的身影。这需要机器人能处理多种家务、适应不同家庭环境(一定的泛化性),更要极度耐用且安全。

4. 工业生产:被寄予厚望的 "深水区"

工业场景相对封闭,非从业者可能难以及时判断进度:

机会:对传统机器难做(如处理线束、布料等柔性物体,或者电子设备精密组装中需要精细力控)或人工太贵、缺工的环节,更通用的具身智能机器人有渗透的机会。

挑战:在成熟的工业门类中,人形机器人面临着 "专机"、传统工业机器人的竞争。现有方案在负载、精度和生产节拍上,短期内超过人形机器人。

工业领域还存在有趣的 "三赢" 潜规则:具身公司向供应链供应商承诺订单,供应商反手买入具身机器人并在二级市场通过相关概念拉升股价。这可能会让机器人在并未真正达到可用状态时就销量先行。

5. 陪伴与娱乐:具身与 AI 硬件之间

陪伴需求不需要极高的智能和任务规划。这类产品的逻辑更接近消费电子:不讲长远的技术故事,直接靠销量和用户口碑说话。它们不需要等待具身智能下一阶段的突破,而是靠现有技术的成熟组合快速回本,再反哺长期研发。

宇树的消费级机器狗 Go1 累计销量已达数万台。而由地平线前副总裁余轶南等人创立的维他动力(Vbot),其超能机器狗在 26 年 1 月的预售期内拿到了 6540 台订单。不过这些订金在锁单前可退,到 26 年 3 月正式锁单并开启交付时,能反映更实际的需求。

我们接触的很多从业者都预言,26 年,具身领域会进入规模化应用落地元年。接下来的 10 个月,我们会看到,这更多是一种期待,还是真的是一个判断。

AGI

Lecun

LeCun: https://www.bilibili.com/video/BV1b1ycYTECU

视频其中一个核心思想是"预测能力的本质是我们找到我们观察的事物的良好表征",事实上现在人类做机器学习的工作大部分是在 寻找表征、优化表征。

最近一段时间伴随LLM出现,技术领域的发展不外乎这两种:1)利用LLM学到的表征去做一些事情;2)让LLM学会更多表征。

  • Lecun的Insight:需要视觉信息训练
    • 反驳"视觉信息冗余"
      • 视神经纤维 1byte/s 已经相比视网膜光传感器有1/100的压缩比了
        • 6000w-1e8光传感器
        • 100w神经纤维
      • self-supervised learning需要冗余信息才能学好
        • 高度压缩==随机 -> 学不好

image-20241019022443123

  • Objective-Driven AI
    • 转化为优化问题,让决策output接近objective,需要先优化perception
    • optimization-based AI
      • 有zero-shot能力
      • search/plan

image-20241019023330137

image-20241019163724135

  • 系统

    • Model Predictive Control(MPC)
      • using gradient-based method, graph search, MCTS, DP, ...
    • 分层的planning,world model预估级联
  • 训练:

    • 观察婴儿对世界模型的认知路径,可以启发各种属性的认知顺序和难度(比如对重力的认知)
    • generative + self-supervised行不通

image-20241019165227218

  • Joint Embedding Predictive Architecture
    • 预测能力的本质是我们找到我们观察的事物的良好表征
      • e.g. 电商场景下的类目体系,类目是对商品的向上一层的抽象表征

image-20241019165308598

image-20241019165600928

image-20241019171905634

image-20241019172914244

  • VICReg
    • 先扩维再正则化

image-20241019173438149

  • Video-JEPA
    • 蒸馏防止collapse

image-20241019173516379

其它

  • 豆包大模型视觉 https://zhuanlan.zhihu.com/p/5761953085

    • 尽管Scaling在Sora上取得成功,但不足以使视频生成模型真正理解并泛化应用基本的物理定律。

      • 模型仅在训练数据分布内表现良好,分布外表现较差,不过Scaling对组合泛化(需组合训练时已熟悉的概念或对象)有效;
      • 模型无法抽象出一般规则,而是试图模仿最接近的训练示例;
      • 当模型参考训练示例时,甚至存在顺序偏好:颜色 > 大小 > 速度 > 形状;
    • 训练数据分布内(in-distribution):训练数据和测试数据来自同一分布,表现良好

    • 训练数据分布外(out-of-distribution):模型在面对从未见过的新场景时,是否能够将已学过的物理定律应用到未知的情境,表现不佳

    • 组合泛化(combinatorial generalization):介于前两者之间,训练数据已包含了所有概念或物体,但这些概念、物体并未以所有可能的组合或更复杂的形式出现,Scaling有效

    • 视频模型具有三种基本组合模式,分别为:

      • 属性组合
      • 空间组合(多个物体不同运动状态)
      • 时间组合(不同的时间点多个物体的不同状态)
    • 视频生成的Scaling Law应当侧重于增加组合多样性,而不仅仅是扩大数据量

Interpretability

Transformer 的可解释性

https://arxiv.org/pdf/2203.04212

https://www.youtube.com/@illcscience7479/videos

Language models can explain neurons in language models: 自解释

  • 步骤:

    • GPT-4解释某个GPT-2神经元的行为
    • 用GPT-4模拟这一行为
    • 比较并打分
  • OpenAI 共让 GPT-4 解释了 GPT-2 中的 307200 个神经元,其中大多数解释的得分很低,只有超过 1000 个神经元的解释得分高于 0.8。

  • 三种提高解释得分的方法:

    • 对解释进行迭代,通过让 GPT-4 想出可能的反例,根据其激活情况修改解释来提高分数。
    • 使用更大的模型来进行解释,平均得分也会上升。
    • 调整被解释模型的结构,用不同的激活函数训练模型。
  • https://github.com/openai/automated-interpretability

  • 传统的视觉解释方法不能scale well

可解释性在电商搜索的应用

Interpretability在电商场景的潜在应用 https://www.vantagediscovery.com/post/the-future-of-e-commerce-is-ai-powered-and-interpretable

  • Hyper-Personalized Product Discovery

    • Scenario: An e-commerce platform wants to move beyond basic recommendation algorithms and create truly personalized shopping experiences that resonate with individual customers. They need to understand not just what products customers interact with, but the underlying reasons and preferences driving their choices.
    • Solution: By using SAEs to analyze the LLM activations associated with user browsing behavior, purchase history, and product interactions (e.g., reviews, ratings, wishlists), the platform can extract nuanced features representing individual preferences and decision criteria. For example, features might emerge for "aesthetic style," "sustainability concerns," "value for money," "brand affinity," or specific functional requirements.
    • image-20241009174406858
  • Optimized Merchandising and Assortment

    • Scenario: A retailer using an e-commerce platform wants to make data-driven decisions about inventory management, product assortment, and merchandising strategies. They need to understand which products are resonating with customers, which attributes are driving demand, and how to optimize pricing and promotions for maximum profitability.
    • Solution: By applying SAEs to analyze the LLM activations linked to product sales data, customer reviews, and market trends, the platform can identify crucial features influencing purchasing decisions. These might include features like "price sensitivity," "seasonal demand," "regional preferences," or "emerging trends" related to specific product attributes.
  • Enhanced Explainable Search

    • Scenario: A customer searches for "running shoes" but is dissatisfied with the results, feeling they are too generic.
    • Solution: The platform could use SAEs to analyze the search query's representation in the LLM's activation space. By identifying the activated features, they could provide an explanation to the customer, like "We are showing you popular running shoes based on your location and browsing history." Additionally, the platform could offer "steering" options based on other relevant features. For example, they could suggest refining the search by "cushioning," "terrain," or "price range."

幻觉

《Lost in the middle: How language models use long contexts》

安全 & 伦理

仅一天就被外媒封杀 前谷歌CEO到底说了... https://v.douyin.com/iBttgjpb/

Prompt安全

AI战争

  • 美国白鹤计划crane war
    • 机器人/无人机摧毁整个军队理论(坦克、炮兵、迫击炮),让地面进攻成为不可能
    • 美国能源不足,加拿大发电,阿拉伯投资

AI安全

  • 关键问题:如何在一个学习了的系统中检测危险(比如混合某些化合物),并且你无法直接询问它这些内容

    • 解决方案:设定一个阈值,超过了向政府报告
  • 火山引擎(BytePlus)ModelArk Content Pre-filter

    • 功能:内容预过滤系统,用于检测输入提示词和输出补全中的特定类别的潜在风险内容
    • 过滤类别
      • Minor Safety:涉及儿童和未成年人隐私侵犯、身心健康伤害的内容
      • Hate Speech and Hateful Content:包括但不限于种族歧视、国籍歧视、民族歧视、性取向、外貌、身体残疾等
      • Nudity, Sexual, and Graphic Content:包括但不限于色情、虐待、暴力、自残等
      • Misinformation:明显具有欺骗性且与事实严重不符的内容
    • 如何启用/禁用
      • 在 ModelArk 平台的导航栏中点击 Online Inference,进入 Custom inference endpoint 界面
      • 在该界面中可以启用或禁用 Content Pre-filter System 功能
    • 重要说明
      • 即使禁用此功能,我们的服务仍会维护基线内容安全政策,努力为每个用户提供积极的使用环境
      • 调用 API 时,Response 中会返回一个字段,帮助用户确定模型生成的内容是否被过滤(finish_reason 字段为 "content_filter")
    • 参考文档

AI政治

  • 对民主的威胁-虚假信息-aigc
    • 尝试解决tiktok问题:平等时间规则(总统候选人的内容播出时间平等)
  • 去中心化的思想构建未来的AI安全:https://mp.weixin.qq.com/s/K1gbW1aIkwl8aLzkD9nYnQ
    • 比特币:攻击收益远小于攻击成本
    • 以生态著称的公链以太坊:虽然秘钥也是几十位,但是系统就太复杂了,各种二层技术、跨链桥等带来了很多漏洞,以至于网络攻击不断,就是因为攻击收益大于攻击成本
    • 方案:确权,实名,竞争

AI伦理

  • 算法演变到最后会扩大"out of rage",因为冲突带来流量
  • 关于丢失工作:需要高等教育的工作没事,因为这些人会和系统协作

下一个学习范式

来源:晚点AI 整理时间:2026-02-16

目前的范式:用海量数据做预训练;用更少、但质量更高的、面对特定任务的数据做监督微调或强化学习的后训练。

热门研究方向:持续学习、在线学习、世界模型等。

AI for Science:用 AI 加速科学发现

来源:晚点聊第 140 期 - 深势科技张林峰、孙伟杰专访 整理时间:2026-02-16

AI for Science 是一个在大语言模型热潮之前便已开始的方向。深势科技(DP Technology)的经历恰好涵盖了该领域的几种核心探索。

核心探索方向

方向一:加速第一性原理计算

背景

  • 第一性原理计算有确定的物理公式:薛定谔方程(Schrödinger Equation)、密度泛函理论(DFT)和分子动力学方程等
  • 这些计算对生化环材领域至关重要
  • 难点:计算复杂度极高,难以从微观尺度跨越到介观或宏观尺度(从单个分子到整体材料属性)

DeePMD 方案

  • 2016 年前后,张林峰在普林斯顿读博期间开发
  • 通过机器学习找到了一种在不损失精度的前提下大幅提升计算效率的方法
  • 深势科技随后据此推出了药物研发计算平台 Hermite

方向二:生成式 AI 解决特定科学问题

典型代表

  • AlphaFold(DeepMind):获得诺贝尔奖,用于预测蛋白质结构
  • Uni-Fold(深势科技):同一方向的模型

深势科技的实践脉络

深势科技的两位创始人张林峰与孙伟杰完整讲述了他们亲历的、用 AI 加速科学发现的发展脉络。这个领域在大语言模型热潮之前便已开始,深势的经历恰好涵盖了该领域的几种核心探索。