Sentence Transformers 的 packed 位置编号对 RoBERTa 不兼容

tomaarsen · x · 2026-07-23

从 v5.5.0 起,Sentence Transformers 在启用 flashattention2 时会把纯文本 batch pack 成单序列。

问题出在 packed 后的 positionids:它会在每段文本里都从 0 重启,但 RoBERTa 家族模型期望的是 paddingidx + 1。结果是每个 token 都读到了偏移的位置嵌入,表现会悄悄变差,但不会崩溃。

所属事件:Flash Attention 致 BGE-M3 等模型掉点及底层原因(2 条相关)→

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →