Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Compress Then Adapt? No, Do It Together via Task-aware Union of Subspaces

About

Adapting large pretrained models to diverse tasks is now routine, yet the two dominant strategies of parameter-efficient fine-tuning (PEFT) and low-rank compression are typically composed in sequence. This decoupled practice first compresses and then fine-tunes adapters, potentially misaligning the compressed subspace with downstream objectives and squandering a global parameter budget. To overcome this limitation, we introduce JACTUS (Joint Adaptation and Compression with a Task-aware Union of Subspaces), a single framework that unifies compression and adaptation. From a small calibration set, JACTUS estimates input and pre-activation gradient covariances, forms their orthogonal union with the pretrained weight subspace, performs a projected low-rank approximation inside this union, allocates rank globally by marginal gain per parameter, and trains only a compact core matrix. This explicitly mitigates the potential misalignment between the compressed subspace and downstream objectives by coupling the directions preserved for compression with those required for adaptation, yielding a deployable low-rank model that avoids retaining full frozen weights while enabling fast and robust tuning. On vision, JACTUS attains an average 89.2% accuracy on ViT-Base across eight datasets at 80% retained parameters, surpassing strong 100% PEFT baselines (e.g., DoRA 87.9%). On language, JACTUS achieves an 80.9% average on Llama2-7B commonsense QA at the same 80% retained-parameter budget, outperforming 100% PEFT (e.g., DoRA 79.7%) and exceeding prior compress-then-finetune pipelines under the same ratained-parameter budget. We will release code.

Jingze Ge, Yun Liu, Xue Geng, Wanqi Dong, Wang Zhe Mark, Min Wu, Xulei Yang• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningGSM8K
Accuracy62.5
1398
Image ClassificationDTD--
599
Image ClassificationRESISC45--
472
Image ClassificationCIFAR-100--
204
Image ClassificationFGVC Aircraft
Top-1 Acc71.5
112
Image ClassificationStanford Cars
Top-1 Accuracy87.8
98
Image ClassificationEuroSAT
Top-1 Accuracy99.3
90
Image ClassificationOxford-IIIT Pet
Top-1 Accuracy95.2
75
Commonsense Question AnsweringCommonsense QA
BoolQ Accuracy70.7
29
Image ClassificationVision Classification Suite (CIFAR-100, EuroSAT, RESISC45, Stanford Cars, FGVC Aircraft, DTD, CIFAR-10, Oxford-IIIT Pets) (test)
CIFAR-100 Accuracy95.2
20
Showing 10 of 11 rows

Other info

Follow for update