AINeutralarXiv – CS AI · 7h ago6/10
🧠
TAPIOCA: Why Task- Aware Pruning Improves OOD model Capability
Researchers demonstrate that task-aware layer pruning improves model performance on out-of-distribution (OOD) data while providing no benefits for in-distribution data. The improvement occurs because pruning removes layers that distort the task-adapted geometric representation, realigning OOD inputs with the model's learned task geometry.