SHIFT-LLM: Distribution Shift Correction in Depth-Pruned LLMs
DGX agentarXiv:2608.25068v1 Announce Type: new Abstract: Depth pruning removes entire Transformer blocks to reduce the inference cost of large language models, but disrupts the hidden-state distributions expec