论文研究
arxiv.org ·
耦合校准与学习:在无目标域奖励反馈下缓解大语言模型蒸馏中的教师偏差
该论文提出 Coupled Calibration and Learning(CCL)蒸馏算法,针对协变量偏移下教师模型在目标问题上可靠性不确定、且无目标域奖励反馈的场景。方法通过 token 级分支将教师校准与学生更新耦合:每轮仅用源问题奖励反馈校准教师,再用校准后的教师训练学生,更新后的学生反过来影响后续校准。作者在自回归策略框架下证明,学生相对 oracle 学生的期望平均 KL 散度以多项式速率收敛到零,并证明其与正则化直接匹配方法存在分离——后者相对 oracle 学生的误差可能始终有界远离零。
AI推荐理由 arXiv 正式提交的机器学习论文(cs.LG/cs.AI),作者含 MIT 的 David Simchi-Levi,摘要给出明确方法(CCL:用源域奖励反馈校准教师、再训练学生,迭代耦合)与理论结果(学生到 oracle 学生的期望平均 KL 散度以多项式速率收敛到零,并给出与正则化直接匹配的分离性结论)。新信息在于:在无目标域奖励反馈、存在协变量偏移时缓解教师偏差,并给出收敛性证明。局限是仅有理论分析,摘要未提供实验或实际效果证据,且目标域无奖励反馈的假设较强。
72
正文
Computer Science > Machine Learning
arXiv:2609.17474 (cs)
[Submitted on 15 Sep 2026]
Title:Coupled Calibration and Learning: Mitigating Teacher Bias in LLM Distillation without Target-Domain Reward Feedback
Authors:Haichen Hu, Yuheng Zhang, David Simchi-Levi
View a PDF of the paper titled Coupled Calibration and Learning: Mitigating Teacher Bias in LLM Distillation without Target-Domain Reward Feedback, by Haichen Hu and 2 other authors
View PDF
HTML (experimental)
Abstract:Large language model (LLM) distillation aims to transfer the capabilities of a powerful teacher to a smaller student. Direct imitation, however, can also transfer the teacher's systematic bias and errors. This challenge is particularly pronounced under covariate shift, when the teacher's reliability on target questions is uncertain and target-domain reward feedback is unavailable. We propose Coupled Calibration and Learning (CCL), an LLM distillation algorithm that couples teacher calibration with student updates through token-level branching, using reward feedback only on source questions. Each iteration calibrates the teacher using source feedback and then uses the calibrated teacher to train the student on target questions. The updated student, in turn, informs subsequent calibration. In an autoregressive policy framework, we prove that the output student's expected average Kullback-Leibler divergence to the oracle student converges to zero at a polynomial rate in the number of iterations. The oracle maximizes the true reference-regularized target reward within the student class, which need not represent the unrestricted optimal policy. Our analysis quantifies the progress of projected student gradient updates while controlling the error in teacher calibration. We further establish a separation from regularized direct matching: its error relative to the oracle student can remain bounded away from zero even when the teacher achieves higher regularized target reward than every student policy. These results demonstrate that LLM distillation can overcome persistent teacher bias and recover the optimal student through coupled calibration and learning, without target-domain reward feedback.
Subjects:
Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Statistics Theory (math.ST); Machine Learning (stat.ML)
Cite as:
arXiv:2609.17474 [cs.LG]
(or
arXiv:2609.17474v1 [cs.LG] for this version)
https://doi.org/10.48550/arXiv.2609.17474
Focus to learn more
arXiv-issued DOI via DataCite (pending registration)
Submission history
From: Haichen Hu [view email]
[v1]
Tue, 15 Sep 2026 17:15:40 UTC (102 KB)
Full-text links:
Access Paper:
View a PDF of the paper titled Coupled Calibration and Learning: Mitigating Teacher Bias in LLM Distillation without Target-Domain Reward Feedback, by Haichen Hu and 2 other authors
View PDF
HTML (experimental)
TeX Source
view license
Current browse context:
cs.LG
new
|
recent
| 2026-09
Change to browse by:
cs
cs.AI
math
math.ST
stat
stat.ML
stat.TH
References & Citations
NASA ADS
Google Scholar
Semantic Scholar
export BibTeX citation
Loading...
BibTeX formatted citation
×
loading...
Data provided by:
Bookmark
Bibliographic Tools
Bibliographic and Citation Tools
Bibliographic Explorer Toggle
Bibliographic Explorer (What is the Explorer?)
Connected Papers Toggle
Connected Papers (What is Connected Papers?)
Litmaps Toggle
Litmaps (What is Litmaps?)
scite.ai Toggle
scite Smart Citations (What are Smart Citations?)
Code, Data, Media
Code, Data and Media Associated with this Article
alphaXiv Toggle
alphaXiv (What is alphaXiv?)
Links to Code Toggle
CatalyzeX Code Finder for Papers (What is CatalyzeX?)
DagsHub Toggle
DagsHub (What is DagsHub?)
GotitPub Toggle
Gotit.pub (What is GotitPub?)
Huggingface Toggle
Hugging Face (What is Huggingface?)
ScienceCast Toggle
ScienceCast (What is ScienceCast?)
Demos
Demos
Replicate Toggle
Replicate (What is Replicate?)
Spaces Toggle
Hugging Face Spaces (What is Spaces?)
Spaces Toggle
TXYZ.AI (What is TXYZ.AI?)
Related Papers
Recommenders and Search Tools
Link to Influence Flower
Influence Flower (What are Influence Flowers?)
Core recommender toggle
CORE Recommender (What is CORE?)
IArxiv recommender toggle
IArxiv Recommender
(What is IArxiv?)
Author
Venue
Institution
Topic
About arXivLabs
arXivLabs: experimental projects with community collaborators
arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.
Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.
Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.
Which authors of this paper are endorsers? |
Disable MathJax (What is MathJax?)