Unifying and Beyond the Theory of Pervasive Capability Degradation in LLM Post-training SFT from Feature Interference
Organizations: College of Computing and Data Science, Nanyang Technological University · School of Computer Science and Engineering, The Chinese University of Hong Kong
Abstract
Pervasive Capability Degradation (PCD) refers to the phenomenon in which supervised fine-tuning (SFT), while adapting a large language model (LLM) to a target task, often erodes the general capabilities acquired during pre-training. Prevailing explanations attribute PCD to macroscopic distribution shift and therefore lack microscopic manipulability, leaving existing mitigation methods without a complete theoretical foundation. In this paper, we first develop a bottom-up geometric theory of PCD grounded in feature interference under superposition: narrow-domain SFT data disrupt the almost-orthogonal feature structure formed during pre-training, and the resulting interference drives activations across decision hyperplanes in the output Laguerre-Voronoi geometry, flipping predictions on general tasks. This theory unifies existing methods into six paradigms and reveals both the source of their efficacy and their theoretical limits. We then prove that cross-entropy under one-hot supervision cannot resolve PCD on its own, and accordingly propose the AEGIS loss, which directly constrains the drift of activation margins along all competing token directions, thereby decoupling interference suppression from task adaptation. Across 3 model families and 14 SFT tasks, AEGIS outperforms representative methods of every paradigm on target tasks and pervasive capabilities simultaneously, at only about 1.3 times the computational cost of standard SFT. Its generality, scalability, and mechanistic interpretability offer a new perspective on fully resolving PCD.
Figures & tables
| Notation | Domain / Dimension | Description |
| 1. Model Architecture & Optimization | ||
| Parameter space | Model parameter set | |
| Dataset | Dataset of training samples | |
| Total number of samples in a dataset or feature set | ||
| Set, | Output vocabulary set and vocabulary size ( ) | |
| Tensor / Vector | General model weight matrices and bias vectors | |
| Model | Method | CoLA | SST-2 | MRPC | STS-B | QQP | MNLI | QNLI | RTE | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| MCC | P-Acc. | Acc. | P-Acc. | Acc./F1 | P-Acc. | Corr. | P-Acc. | Acc./F1 | P-Acc. | M-Acc. | P-Acc. | Acc. | P-Acc. | Acc. | P-Acc. | ||
| Mistral-7B | Vanilla | 0.435 | 0.512 | 0.965 | 0.519 | 0.884 | 0.493 | 0.875 | 0.522 | 0.905 | 0.521 | 0.884 | 0.524 | 0.935 | 0.537 | 0.835 | 0.527 |
| PSFT | 0.449 | 0.514 | 0.977 | 0.528 | 0.891 | 0.485 | 0.886 | 0.518 | 0.892 | 0.515 | 0.892 | 0.519 | 0.939 | 0.542 | 0.854 | 0.525 | |
| SMART | 0.447 | 0.519 | 0.971 | 0.537 | 0.889 | 0.491 | 0.881 | 0.526 | 0.906 | 0.524 | 0.888 | 0.514 | 0.928 | 0.539 | 0.869 | 0.534 | |
| KnOTS | 0.385 | 0.437 | 0.972 | 0.462 | 0.871 | 0.473 | 0.864 | 0.507 | 0.875 | 0.508 | 0.867 | 0.492 | 0.914 | 0.504 | 0.812 | 0.519 | |
| DARE | 0.398 | 0.463 | 0.947 | 0.465 | 0.865 | 0.466 | 0.859 | 0.501 | 0.869 | 0.486 | 0.871 | 0.488 | 0.905 | 0.511 | 0.819 | 0.504 | |
| Model | Method | GSM8K | CodeAlpaca | Code-Bagel-Hermes | Infinity-Instruct-Code | Evol-Instruct | LogiQA2 | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Acc. | P-Acc. | Pass@1 | Pass@5 | Pass@10 | P-Acc. | Token Acc. | P-Acc. | Token Acc. | P-Acc. | Win Rate | P-Acc. | Acc. | P-Acc. | ||
| Mistral-7B | Vanilla | 0.375 | 0.391 | 0.085 | 0.131 | 0.145 | 0.375 | 0.445 | 0.381 | 0.488 | 0.421 | 0.467 | 0.423 | 0.612 | 0.445 |
| PSFT | 0.374 | 0.399 | 0.084 | 0.134 | 0.151 | 0.396 | 0.416 | 0.385 | 0.495 | 0.422 | 0.461 | 0.421 | 0.618 | 0.432 | |
| SMART | 0.403 | 0.396 | 0.098 | 0.141 | 0.151 | 0.378 | 0.394 | 0.389 | 0.492 | 0.415 | 0.472 | 0.425 | 0.614 | 0.438 | |
| KnOTS | 0.326 | 0.357 | 0.064 | 0.117 | 0.123 | 0.326 | 0.384 | 0.354 | 0.457 | 0.372 | 0.455 | 0.394 | 0.547 | 0.405 | |
| DARE | 0.354 | 0.361 | 0.083 | 0.129 | 0.144 | 0.354 | 0.397 | 0.329 | 0.443 | 0.388 | 0.439 | 0.388 | 0.463 | 0.411 | |
| Model | Method | Total Time (GPU h) | Total FLOPs | MFU | HFU | Peak Mem. (GiB) |
|---|---|---|---|---|---|---|
| Qwen2.5-3B | Vanilla | 1.69( ) | 23.5% | 39.2% | 59.9 | |
| PSFT | 25.3 | 3.13( ) | 19.7% | 38.4% | 75.5 | |
| DiFT | 1.75( ) | 23.6% | 42.7% | 61.2 | ||
| AEGIS | 2.13( ) | 12.0% | 20.0% | 60.8 | ||
| Qwen2.5-7B | Vanilla | 3.85( ) | 15.6% | 26.1% | 147.7 | |
| PSFT | 6.72( ) | 13.6% | 28.4% | 199.8 |
| Method | RTE | GSM8K | CodeAlpaca | ||||||
|---|---|---|---|---|---|---|---|---|---|
| Cross. | New | Rec. | Cross. | New | Rec. | Cross. | New | Rec. | |
| Base Model | 18.569 | – | – | 18.569 | – | – | 18.569 | – | – |
| Vanilla | 16.668 | 5.700 | 8.659 | 4.373 | 2.321 | 5.748 | 5.915 | 1.941 | 4.516 |
| PSFT | 16.874 | 5.749 | 8.547 | 4.699 | 2.342 | 5.760 | 5.937 | 1.936 | 4.564 |
| SMART | 16.649 | 5.798 | 8.967 | 4.830 | 2.336 | 5.722 | 5.925 | 1.957 | 4.529 |
| KnOTS | 16.782 | 5.830 | 8.296 | 4.667 | 2.487 | 5.348 | 5.916 | 1.963 | 4.262 |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | #Samples | Evaluation | #Eval. Samples | Metric |
|---|---|---|---|---|
| CoLA | 9,594 | CoLA | 1,043 | MCC |
| SST-2 | 68,221 | SST-2 | 872 | Accuracy |
| MRPC | 4,076 | MRPC | 408 | Accuracy / F1 |
| STS-B | 7,249 | STS-B | 1,500 | Pearson / Spearman |
| QQP | 404,276 | QQP | 40,430 | Accuracy / F1 |
| MNLI | 402,517 | MNLI | 9,815 | Matched accuracy |
| Dataset | Prompt Template | Input Case | Output Case (Label) |
|---|---|---|---|
| GSM8K | Question: {question}\nAnswer: | Natalia sold clips to 48 of her friends in April, and then she sold half as many clips in May. How many clips did Natalia sell altogether in April and May? | ␣Natalia sold 48/2 = <<48/2=24>>24 clips in May.\nNatalia sold 48+24 = <<48+24=72>>72 clips altogether in April and May.\n#### 72 |
| CoLA | Sentence: "{sentence}"\nIs this sentence grammatically acceptable English? Answer acceptable or unacceptable.\nAnswer: | The sailors rode the breeze clear of the rocks. | ␣acceptable |
| SST-2 | Sentence: "{sentence}"\nWhat is its sentiment? Answer positive or negative.\nSentiment: | it ’s sweet and fluffy at the time , | ␣positive |
| MRPC | Sentence 1: "{s1}"\nSentence 2: "{s2}"\nDo these sentences have the same meaning? Answer yes or no.\nAnswer: | s1: Slow-moving , drenching Tropical Storm Henri doused an already soaked Florida Friday , … s2: … on Friday , pushing heavy rains into areas where lakes and rivers were full to overflowing . | ␣yes |
| STS-B | Sentence 1: "{s1}"\nSentence 2: "{s2}"\nRate their semantic similarity from 0.0 (unrelated) to 5.0 (equivalent). Output one number.\nScore: | s1: A man with a hard hat is dancing. s2: A man wearing a hard hat is dancing. | ␣5.0 (trained with the MSE of the expected value over number tokens instead of cross-entropy) |
| QQP | Question 1: "{q1}"\nQuestion 2: "{q2}"\nAre these duplicate questions with the same intent? Answer yes or no.\nAnswer: | q1: Why are African-Americans so beautiful? q2: Why are hispanics so beautiful? | ␣no |
| Method | Mix-up Group 1 | Mix-up Group 2 | Mix-up Group 3 | ||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| SST-2 | GSM8K | CodeAlpaca | LogiQA2 | P-Acc. | RTE | GSM8K | Code-Bagel-Hermes | Evol-Instruct | P-Acc. | CoLA | GSM8K | CodeAlpaca | Evol-Instruct | P-Acc. | |
| Vanilla | 0.938 | 0.354 | 0.115 | 0.535 | 0.795 | 0.343 | 0.417 | 0.429 | 0.522 | 0.405 | 0.349 | 0.114 | 0.429 | 0.539 | |
| PSFT | 0.929 | 0.361 | 0.111 | 0.539 | 0.801 | 0.351 | 0.419 | 0.435 | 0.526 | 0.394 | 0.356 | 0.116 | 0.436 | 0.541 | |
| SMART | 0.941 | 0.355 | 0.118 | 0.537 | 0.798 | 0.348 | 0.412 | 0.436 | 0.523 | 0.409 | 0.348 | 0.113 | 0.439 | 0.535 | |
| KnOTS | 0.925 | 0.307 | 0.094 | 0.491 | 0.764 | 0.324 | 0.395 | 0.405 | 0.475 | 0.359 | 0.314 | 0.105 | 0.405 | 0.501 | |
| DARE | 0.918 | 0.339 | 0.098 | 0.509 | 0.778 | 0.334 | 0.408 | 0.401 | 0.488 | 0.387 | 0.316 | 0.101 | 0.411 | 0.514 | |
| Rank | Base | MRPC | CodeAlpaca | GSM8K | |||
|---|---|---|---|---|---|---|---|
| Vanilla | AEGIS | Vanilla | AEGIS | Vanilla | AEGIS | ||
| 1 | </s> <0x0A> (11784) | </s> ␣yes (4327) | . </s> (1578) | . , (3330) | , ␣and (172) | ␣answer ␣final (4154) | . , (218) |
| 2 | </s> ␣ (5768) | </s> ␣no (3474) | <0x0A> </s> (1473) | 1 2 (3173) | . ␣and (156) | ␣a ␣the (2250) | , ␣and (125) |
| 3 | </s> . (5151) | </s> yes (1927) | , </s> (1445) | 1 4 (2150) | . , (52) | ␣The </s> (1156) | ␣the ␣a (115) |
| 4 | </s> - (4044) | ␣( </s> (1259) | , ␣and (1342) | 3 4 (2121) | <0x0A> </s> (42) | 1 2 (659) | . ␣and (93) |
| 5 | </s> ( (3879) | </s> no (729) | ␣ </s> (1330) | ␣a ␣the (2119) | , . (39) | 4 2 (237) | , . (92) |
Language Model Post-Training
Language model post-training adapts pretrained models to follow instructions, reason, use tools, and meet task-specific or human-defined goals.
See all 200 papers