精读笔记
Problem Setting
[Towards Metric-Agnostic Trajectory Forecasting]（arXiv preprint / 2026）
这篇论文实际处理的是 trajectory forecasting 里“预测分布”和“benchmark 输出”混在一起的问题。标准任务看似是给 K 条未来轨迹，但真正要建模的是未来行为的连续、多模态、不确定分布；K 条轨迹只是 benchmark 规定的压缩格式。困难点在于这个压缩格式本身带有强 metric bias：minFDE 奖励至少一个 endpoint 靠近 GT，soft mAP / miss rate 奖励覆盖不同窗口，二者天然冲突。
以前方法卡住的地方不是架构表达力不足，而是训练目标已经被评测协议塑形。WTA、k-means、NMS、metric-specific matching 都在把模型往某个 benchmark 的最优 reporting strategy 上推。结果是模型输出看起来 SOTA，但很难说它是否学到了真实可迁移的不确定性。本文的核心问题就是：能否先学一个 metric-agnostic predictive distribution，再把不同 metric 的优化下放到评测时的 decision policy。
Motivation
作者的核心观察很直接：Argoverse 2 上的强方法多围绕 Brier-minFDE / endpoint distance 设计，Waymo 上的强方法多围绕 soft mAP 的窗口覆盖设计。它们不是在同一个目标上做得更好，而是在优化不同 metric 所鼓励的不同行为。distance-based metric 允许甚至鼓励多个 mode 聚在主模态附近；window-based metric 则惩罚或忽略冗余 mode，鼓励 endpoint 分散覆盖。
已有路线缺的是一个把“distribution learning”从“metric reporting”里剥离出来的框架。只要训练时直接对齐 benchmark，模型改进和 metric hacking 就不可分。作者因此选择把 benchmark 指标解释为 downstream decision problem：如果预测分布可信，那么不同指标下应该可以通过不同 policy 从同一个分布中抽取最优 K 个代表点。
Core Idea
论文真正的核心不是 DONUT-NLL，也不是某个采样技巧，而是一个 decision-theoretic reframing：forecasting model 输出完整预测分布，benchmark submission 是对该分布的一次 metric-specific action selection。TraDiE policy 就是把概率分布映射为 K 条轨迹和置信度的策略，其目标是在“GT 由模型分布产生”的假设下最大化对应 metric。
这个想法改变了建模方式：以前 K 个 mode 是模型训练的直接对象，现在 K 个 mode 是分布的后验摘要；以前训练目标内嵌 metric bias，现在 metric bias 被显式放到测试时 policy 中。新增的 inductive bias 是校准优先：模型只有在概率质量放得合理时，policy 才能选出对真实数据也有效的 endpoints。相比 WTA/NMS，这种组织方式更 generalizable，因为同一分布可被多个 policy 消费；但它也更依赖分布本身是否真实。
Method
方法层面只需要抓住几个机制。
第一，TraDiE 的 distance policy 把 minFDE 视为分布下的 expected min distance minimization。它从 endpoint distribution 采样，把 K 个 endpoint 当作自由变量优化，使它们成为该分布下的 K-median 式代表点。它解决的是：给定一个连续分布，如何生成最适合 minFDE 的有限摘要。
第二，window policy 把 soft mAP / mAP / miss rate 视为覆盖问题。它用采样点近似可能 GT，每个可能 GT 对应一个评测窗口，然后贪心选择能覆盖最多尚未覆盖窗口的 endpoint，并用被覆盖概率作为 confidence。它解决的是：给定分布，如何避免冗余 endpoint，并把高概率 true positive 排到前面。
第三，DONUT-NLL 用 Traj-NLL / Step-NLL 替代 DONUT 的 WTA loss。WTA 只更新被选中的最近 mode，本质上会把 mode 学成 metric candidate；NLL 则要求整个 mixture 解释数据分布。Step-NLL 进一步允许 mixture weight 随时间变化，增加表达力，但也削弱 trajectory-level mode 的一致性。
第四，位置分布从 Laplace 扩展到 generalized Gaussian / Gaussian scale mixture。这里的机制意义是让单个 mode 的 tail shape 和 sharpness 可调，减少因为分布族错配导致的 calibration error。这个点重要，但更像支撑分布建模质量的工程性/建模性选择，而不是范式本身。
Key Insight / Why It Works
最重要的 insight 是：benchmark 所需的 K 条轨迹不应被当成模型内部 latent modes，而应被当成从预测分布中导出的 decision set。这个区分非常关键。一个真实未来分布可能需要密集概率质量、长尾、相关不确定性和多尺度 shape；而 K=6 的输出只是一个面向指标的摘要。把两者混为一谈，必然会让模型学到 metric-specific artifacts。
方法有效的原因主要有三层。第一，NLL 迫使模型对完整概率质量负责，而不是只让“最近 mode”对样本负责，因此更容易形成可被采样和再优化的分布。第二，TraDiE 把不同指标的冲突显式推迟到 evaluation-time decision policy，而不是让一个训练目标同时承担互相矛盾的要求。第三，policy 使用额外 test-time compute，把有限 K 输出问题转化为分布下的近似优化；这部分增益不能完全归因于模型变强，而是来自更合理的 test-time decision making。
我认为核心贡献是“distribution-first + metric-specific policy”的分解，而不是 DONUT 本身的改造。Step-NLL 和 generalized Gaussian 是让这个分解在 Waymo 上有效的关键配套，但不是概念上的主贡献。尤其 Step-NLL 的好处可能部分来自更高表达力和更灵活的逐时刻 mixture assignment；它是否真的学到 trajectory-level calibrated distribution，文中未充分说明。
需要直接指出：这不是纯粹 metric-agnostic。训练目标是 metric-agnostic，但 evaluation policy 是 metric-aware，而且显式用 benchmark window / minFDE definition 做优化。因此论文真正主张应理解为“训练分布与指标解耦”，而不是“不使用指标信息”。增益中相当一部分来自 test-time compute 和 metric-aware extraction；这不是缺点，但归因必须清楚。
Relation To Prior Work
它最接近 HOME 的谱系：先学 endpoint distribution，再针对 metric 从分布中抽取有限预测。但 HOME 绑定 raster heatmap、使用非严格概率训练和若干经验 heuristic；本文把这件事抽象成 model-agnostic sampling policy，并用 NLL 训练 normalized distribution。这是实质差异。
和 WTA/Laplace mixture 系方法相比，区别在于 mode 不再只是“谁离 GT 最近谁更新”的候选轨迹，而是概率混合模型的组成部分。和 Waymo 系 NMS 方法相比，区别在于 NMS 是对一批候选 endpoints 做几何去重，而 TraDiE window policy 是在预测分布下近似最大覆盖，confidence 也来自 true-positive probability 的估计。
看似新的部分中，minFDE policy 本质上接近 continuous K-median / quantization，window policy 接近 maximum coverage greedy，NLL mixture training 也不是新东西。真正新增的信息是把这些已有思想组织成一个统一的评测协议：用同一个预测分布，通过不同 policy 适配冲突指标，并把 policy 成败作为分布质量诊断。
Dataset / Evaluation
实验主要在 Waymo motion prediction 上验证，覆盖了 window-based 和 distance-based 指标，并包含 DONUT、MTR、QCNet 三类模型/设置。它较好支持了论文的核心 claim：在同一 benchmark 内，一个分布模型可以通过不同 policy 适配不同指标；policy 对 MTR 有迁移性；QCNet 失败案例说明 policy 能暴露分布错配。
但 evaluation 的外推范围有限。首先，核心结果集中在 Waymo，Argoverse 只给出 Step-NLL 可能过拟合的负面线索，因此跨 benchmark 泛化没有被充分证明。其次，指标都是 offline endpoint-heavy metrics，不能验证预测分布对闭环规划、安全约束或交互建模的实际价值。再次，TraDiE policy 直接针对 benchmark metric 定义优化，因此它验证的是“distribution + metric-aware extraction”在 benchmark 上有效，而不是证明模型输出本身就是 deployment-ready forecast。
SOTA 对比有说服力但也要谨慎读：同一模型用 window policy 和 distance policy 分别提交时拿到不同指标优势，这说明分布可被复用，不等价于一次输出同时满足所有实际下游需求。
Limitation
最大的隐含前提是：模型预测分布足够校准，且 benchmark GT 可以近似看作从该分布采样。一旦分布错配，policy 会忠实优化错误分布，QCNet 的 minFDE 崩溃就是例子。这也意味着 TraDiE 不是补救坏分布的后处理，而是放大分布质量差异的诊断器。
第二个上限是 endpoint-centric。论文的 policies 主要围绕 endpoint metrics，甚至在 Waymo 多 horizon 下通过 rank-wise pairing 组合 endpoint；这不保证轨迹连续、动力学可行，也不保证 mode 在时间上的语义一致。对真实规划而言，K 条不连续或弱连续的 endpoint 摘要价值有限。
第三，Step-NLL 的表达力可能是双刃剑。逐时刻 mixture weight 能提升 endpoint metric，但可能破坏完整轨迹 latent mode 的一致性。文中未充分说明它学到的是 coherent trajectory distribution，还是更像每个时间点的 marginal density 更好。
第四，增益归因不完全干净。模型训练、分布族、更强 policy、test-time sampling/optimization 同时变化；虽然 ablation 做得不少，但“真正来自校准分布”还是“来自更强的 evaluation-time optimizer”没有完全分离。部分收益可以明确视为 test-time compute 和 benchmark-aware decision policy，而非 backbone forecasting 能力提升。
第五，真实 deployment 鸿沟明显。作者也承认 policy 不是为车端部署设计；更深层问题是 planner 需要的是任务相关 utility 下的可执行未来假设，而不是 endpoint metric 的最优代表点。因此该框架更适合作为评测/诊断范式，而不是直接替代预测模块输出。
Takeaway
- 1. 最值得记住的是把 K 轨迹输出从“模型预测对象”降级为“分布摘要”。
- 这对 trajectory forecasting 很重要，也可迁移到其他需要用少量 hypothesis 表达连续不确定性的任务。
- 2. 训练目标和评测指标解耦后，很多已有 design choice 会反转：WTA 在 naive minFDE 下看起来好，但在 distribution-first 评估下不一定好。
- 这提醒后续工作应显式评估 predictive distribution，而不是只比较 submitted trajectories。
一句话总结
这篇论文在 trajectory forecasting 中把“学习未来分布”和“面向指标抽取 K 条轨迹”明确解耦，实质贡献是将 SOTA 竞争从 metric-specific mode engineering 推向 distribution-first forecasting + test-time decision policy 的范式。