Task-Conditional Flow Matching for Balanced Multilingual Text Embedding Adaptation
任务条件流匹配:多语言文本嵌入的自适应平衡训练
🧭 多语言嵌入任务自适应训练范式
LINGO Research Group, Indian Institute of Technology Gandhinagar
提出任务条件流匹配,对翻译任务用流匹配,其他任务用匹配目标,结合教师引导与课程学习,多语言嵌入达 SOTA
📌 工作参考价值(4 点)
- 多任务嵌入微调避免冲突:在处理翻译、检索、分类等混合任务时,可借鉴 TCFM 对不同任务家族采用不同的训练目标,例如翻译用流匹配(适合连续对齐),检索/分类仍用对比学习/交叉熵,避免单一目标造成性能跷跷板。
- 教师引导与课程学习稳定适应:通过冻结原始模型做教师,加蒸馏损失保持旧知识,再配合三阶段课程(先教师引导,再逐步引入新任务),可有效防止灾难性遗忘,适用于业务中持续更新多语言嵌入模型的场景。
- 多语言搜索/商品检索:若业务涉及跨语言商品搜索、query 理解,可借鉴 TCFM 对平行语料(翻译对)使用流匹配直接对齐表示,提升多语言 embedding 的一致性,而不损伤其他任务指标。
- 工程实现参考:流匹配模块轻量,可与对比学习头并存,通过 task-conditioned 开关选择路径;课程学习的分阶段训练策略可直接嵌入现有 ML pipeline,无需大幅改动架构。