Search papers, labs, and topics across Lattice.
This paper explores the concept of loss smoothing to enhance the adaptation of neural networks under distribution shifts, particularly in fine-tuning and reinforcement learning contexts. By interpolating between the source and target training objectives at the onset of adaptation, the method mitigates the abrupt transitions that can distort learned representations. The results demonstrate that loss smoothing consistently improves performance across various tasks, indicating its potential as a broadly applicable strategy for model adaptation.
Gradual transitions in training objectives can significantly enhance model performance during adaptation, preserving valuable learned features.
In settings such as fine-tuning and reinforcement learning, neural networks are often adapted under distribution shift. Standard adaptation methods typically optimize the target objective directly, inducing an abrupt change from the source training objective. This abrupt transition can distort learned representations, including features that may still be useful for the new task. We investigate whether a more gradual transition can improve adaptation. We propose loss smoothing, a simple approach that interpolates between the source and target training objectives at the start of adaptation. This smooth transition helps to preserve useful features from the source distribution while still enabling the model to specialize to the target distribution. Across controlled supervised shifts, pretrained vision adaptation, offline-to-online and online reinforcement learning, and language model fine-tuning, we find that loss smoothing consistently improves performance, suggesting that smoother objective transitions are a broadly useful tool for model adaptation.
Mila