LLM-Enhanced Dual-Branch Learning for Large-Scale Multi-Label Text Classification
Not on the map. Why?
Abstract (arXiv)
Large-scale multi-label text classification assigns a small subset of relevant labels to each document from a vocabulary containing thousands or tens of thousands of candidate labels. Although pretrained language models have improved semantic text representations, most representation-based approaches center their prediction pipelines on a primary encoder or combine auxiliary features within a single ranker. The complementarity between heterogeneous language models therefore remains insufficiently explored. We propose DualMLC, a dual-branch framework that processes the same document through an autoregressive decoder-only language model and a bidirectional encoder. Each branch maintains its own representation pathway and independently estimates relevance scores over the shared label space. DualMLC combines the two score vectors through late logit fusion, allowing shared evidence to reinforce relevant labels and branch-specific evidence to compensate for limitations in the other branch's representation. DualMLC achieves state-of-the-art results on three widely used large-scale multi-label text classification benchmarks. Ablation results further confirm that integrating the heterogeneous predictors produces stronger rankings than either branch alone. The source code is publicly available at https://github.com/huiyegit/DualMLC.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 14 of its 27 references that are in the map. First posted on arXiv 11 Sep 2026.
Builds on
14 of 27 references.
References · 14
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2018AttentionXML: Label Tree-based Attention-Aware Deep Model for High-Performance Extreme Multi-Label Text Classificationcited by 102
- 2016DiSMEC - Distributed Sparse Machines for Extreme Multi-label Classificationcited by 71
- 2021Fast Multi-Resolution Transformer Fine-tuning for Extreme Multi-label Text Classificationcited by 67
- 2019Bonsai -- Diverse and Shallow Trees for Extreme Multi-label Classificationcited by 61
- 2021LightXML: Transformer with Dynamic Negative Sampling for High-Performance Extreme Multi-label Text Classificationcited by 61
- 2018A no-regret generalization of hierarchical softmax to extreme multi-label classificationcited by 52
- 2020PECOS: Prediction for Enormous and Correlated Output Spacescited by 50
- 2020Pretrained Generalized Autoregressive Model with Adaptive Probabilistic Label Clusters for Extreme Multi-label Text Classificationcited by 25
- 2022CascadeXML: Rethinking Transformers for End-to-end Multi-resolution Training in Extreme Multi-label Classificationcited by 12
Built on by
Not cited yet.