Preserving Zero-shot Capability in Supervised Fine-tuning for Multi-label Text Classification

Si-An Chen, Hsuan-Tien Lin, Chih-Jen Lin


Abstract
Zero-shot multi-label text classification (ZMTC) requires models to predict multiple labels for a document, including labels unseen during training. Previous work assumes that models leveraging label descriptions ensures zero-shot capability. However, we find that supervised methods, despite achieving strong overall performance, lose their zero-shot capability during training, revealing a trade-off between overall and zero-shot performance. To address the issue, we propose OF-DE and OF-LAN, which preserve the zero-shot capabilities of powerful dual encoder and label-wise attention network architectures by freezing the label encoder. Additionally, we introduce a self-supervised auxiliary loss to further improve zero-shot performance. Experiments demonstrate that our approach significantly improves zero-shot performance of supervised methods while maintaining strong overall accuracy.
Anthology ID:
2025.findings-naacl.315
Volume:
Findings of the Association for Computational Linguistics: NAACL 2025
Month:
April
Year:
2025
Address:
Albuquerque, New Mexico
Editors:
Luis Chiruzzo, Alan Ritter, Lu Wang
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
5699–5712
Language:
URL:
https://preview.aclanthology.org/landing_page/2025.findings-naacl.315/
DOI:
Bibkey:
Cite (ACL):
Si-An Chen, Hsuan-Tien Lin, and Chih-Jen Lin. 2025. Preserving Zero-shot Capability in Supervised Fine-tuning for Multi-label Text Classification. In Findings of the Association for Computational Linguistics: NAACL 2025, pages 5699–5712, Albuquerque, New Mexico. Association for Computational Linguistics.
Cite (Informal):
Preserving Zero-shot Capability in Supervised Fine-tuning for Multi-label Text Classification (Chen et al., Findings 2025)
Copy Citation:
PDF:
https://preview.aclanthology.org/landing_page/2025.findings-naacl.315.pdf