cs.CLOct 8, 2026

SFT-as-Context Mitigates Forgetting in Supervised Fine-Tuning

Authors: Kenan Tang, Andong Hua, Chengxuan Qian, Saket Tiwari, Yao Qin

Organizations: University of California, Santa Barbara

Abstract

Supervised fine-tuning (SFT) equips large language models (LLMs) with specialized capabilities, but often comes at the cost of forgetting the general capabilities of their parent models (i.e., the pretrained models before fine-tuning). This trade-off is especially limiting for queries that require both specialized and general capabilities. We introduce SFT-as-context, a training-free method in which the parent model uses the SFT model's response as context to answer the query. This allows the parent model to acquire fine-tuned capabilities from the SFT response through in-context learning while preserving its own general capabilities. Across 19 parent-SFT model pairs and 11 benchmarks, SFT-as-context remains close to the SFT models on fine-tuned capabilities, with gaps of only 2.2 and 2.1 percentage points on AIME 2024 and LiveCodeBench and 2.0 macro MAE on NutriBench-English, while staying within 2.2 percentage points of the parent models on general capabilities on average. Remarkably, it can solve queries requiring both fine-tuned and general capabilities, even when neither the parent nor SFT model succeeds alone. This approach also extends beyond parent-SFT pairs: responses from a small open-source SFT model can improve a strong closed-source LLM, outperforming either model alone. Furthermore, we use a Bayesian framework to derive theoretical guarantees that bound the error of SFT-as-context relative to the SFT model on fine-tuned capabilities and to the parent model on general capabilities. In addition, we visualize the attention weights and find that the parent model attends more to useful SFT responses and less to irrelevant ones, suggesting that selective attention helps the parent model use the SFT response through in-context learning.

Explore similar work

CardsList
  1. InfoSFT: Learn More and Forget Less with Information-Aware Token Weighting

    May 14, 2026Mahdi Sabbaghi, George Pappas, Adel Javanmard +1Supervised Fine-TuningAdaptive Loss Weighting

  2. FocuSFT: Bilevel Optimization for Dilution-Aware Long-Context Fine-Tuning

    May 11, 2026Zehua Pei, Hui-Ling Zhen, Xianzhi Yu +3Supervised Fine-TuningSelf-Attention

  3. On-Policy Replay for Continual Supervised Fine-Tuning

    May 28, 2026Yan Chen, Taojie Zhu, Meng Zhang +4Continual Learning for LLMsReplay-Based Continual Learning