Mastering the Minority: An Uncertainty-guided Multi-Expert Framework for Challenging-tailed Sequence Learning
Organizations: Key Laboratory of Cyberspace Big Data Intelligent Security, Ministry of Education; School of Artificial Intelligence, Chongqing University of Post and Telecommunications, Chongqing 400065, China. · School of Computer Science and Technology, East China Normal University, Shanghai, 200062, China. · Chongqing Key Laboratory of Brain-Inspired Cognitive Computing and Educational Rehabilitation for Children with Special Needs, Chongqing Normal University, Chongqing, 401331, China.
Abstract
Imbalanced data distribution remains a critical challenge in sequential learning, leading models to easily recognize frequent categories while failing to detect minority classes adequately. The Mixture-of-Experts model offers a scalable solution, yet its application is often hindered by parameter inefficiency, poor expert specialization, and difficulty in resolving prediction conflicts. To Master the Minority classes effectively, we propose the Uncertainty-based Multi-Expert fusion network (UME) framework. UME is designed with three core innovations: First, we employ Ensemble LoRA for parameter-efficient modeling, significantly reducing the trainable parameter count. Second, we introduce Sequential Specialization guided by Dempster-Shafer Theory (DST), which ensures effective specialization on the challenging-tailed classes. Finally, an Uncertainty-Guided Fusion mechanism uses DST's certainty measures to dynamically weigh expert opinions, resolving conflicts by prioritizing the most confident expert for reliable final predictions. Extensive experiments across four public hierarchical text classification datasets demonstrate that UME achieves state-of-the-art performance. We achieve a performance gain of up to 17.97% over the best baseline on individual categories, while reducing trainable parameters by up to 10.32%. The findings highlight that uncertainty-guided expert coordination is a principled strategy for addressing challenging-tailed sequence learning. Our code is available at https://github.com/CQUPTWZX/Multi-experts.
Figures & tables
| Dataset | Depth | # Train | # Dev | # Test | ||
| WOS | 141 | 2.0 | 2 | 30,070 | 7,518 | 9,397 |
| RCV1-V2 | 103 | 3.24 | 4 | 18,520 | 4,629 | 781,265 |
| AAPD | 54 | 2.41 | 2 | 43,872 | 10,968 | 1,000 |
| BGC | 146 | 3.01 | 4 | 58,800 | 14,700 | 18,394 |
| Model | WOS | RCV1-V2 | AAPD | BGC | NYT | |||||
| Micro-F1 | Macro-F1 | Micro-F1 | Macro-F1 | Micro-F1 | Macro-F1 | Micro-F1 | Macro-F1 | Micro-F1 | Macro-F1 | |
| Hierarchy-Aware Models | ||||||||||
| HiAGM (ACL2020) | 85.11 1.4 | 80.02 0.7 | 83.17 1.8 | 62.35 1.9 | 75.54 2.1 | 58.86 1.2 | 76.61 0.9 | 57.27 0.7 | — | — |
| HTCInfoMax (NAACL2021) | 84.86 0.6 | 79.12 0.2 | 83.26 0.8 | 60.66 1.5 | 77.42 0.7 | 57.72 0.7 | 74.26 0.4 | 56.81 1.0 | — | — |
| HiMatch (ACL2021) | 85.04 0.9 | 80.37 1.3 | 84.42 0.9 | 63.99 1.1 | 76.70 0.3 | 59.86 0.5 | 76.63 0.8 | 57.87 1.0 | — | — |
| Pretrained Language Models | ||||||||||
| Parameters | Description |
| r | LoRA Rank. Default: 64 |
| lr | Learning rate |
| batch | Batch size |
| epoch | Default: 10 |
| early-stop | Epoch before early stop |
| update | Gradient accumulate steps |
| Category | Count | Ours | HiTIN | HJCL | HGCLR | HPT |
| g157 | 1991 | 55.6 5% | 55.25% | 51.28% | 49.42% | 50.77% |
| c16 | 1871 | 58.04 % | 54.62% | 57.96% | 53.82% | 55.41% |
| gwelf | 1818 | 48.01% | 51.65 % | 49.87% | 45.21% | 45.21% |
| e311 | 1658 | 79.79 % | 61.46% | 58.74% | 61.82% | 59.18% |
| c331 | 1179 | 77.26% | 67.77% | 79.55 % | 54.20% | 74.36% |
| e143 | 1172 | 77.39 % | 65.61% | 66.20% | 65.87% | 65.31% |
| Datasets | Top N Tailed Labels | HGCLR | HPT | HiTIN | HJCL | HILL | Ours |
| WOS | N=8 | 28.47 | 27.78 | 37.83 | 41.67 | 42.08 | 46.60 |
| N=16 | 48.67 | 49.04 | 53.16 | 57.67 | 51.78 | 62.66 | |
| N=32 | 61.49 | 65.36 | 61.43 | 71.62 | 68.44 | 74.70 | |
| RCV1-V2 | N=8 | 18.92 | 16.88 | 20.24 | 22.40 | 18.38 | 26.93 |
| N=16 | 41.83 | 41.37 | 47.21 | 43.22 | 46.72 | 50.45 | |
| N=32 | 56.67 | 54.77 | 57.98 | 57.20 | 61.46 | 63.69 |
| Experts ( ) | Tail Micro-F1 | Avg. Conflict ( ) | Last Expert Util. |
| 3 | 76.21 | 0.1724 | 20.26 |
| 4 | 75.32 | 0.1597 | 1.80 |
| 5 | 74.05 | 0.1175 | 1.28 |
| Ablation Study | WOS | RCV1-V2 | AAPD | BGC |
| w/o | 87.52 | 86.61 | 78.86 | 79.66 |
| w/o ( ) | 86.98 | 86.37 | 78.15 | 79.27 |
| w/o ( ) | 85.91 | 85.72 | 77.28 | 78.51 |
| Direct Integration | 87.18 | 85.73 | 78.53 | 79.51 |
| Gated MoE | 86.74 | 86.12 | 77.91 | 79.08 |
| UME | 87.71 | 86.88 | 79.21 | 80.46 |
| Model | WOS | RCV1-V2 | ||
| Micro-F1 | Macro-F1 | Micro-F1 | Macro-F1 | |
| GPT-4 | 59.74 | 37.80 | 56.12 | 36.54 |
| LlaMa-3 | 61.91 | 27.33 | 54.67 | 28.50 |
| UME (Ours) | 87.71 | 82.29 | 86.88 | 69.70 |
| Prompt Template | You are an expert hierarchical text classifier. Your task is to classify the given text into the correct category path from the predefined taxonomy. [Taxonomy Definition] The categories are defined in a hierarchy formatted as: - The symbol “ ” represents the parent-child relationship. - You must output the full path from the root to the specific leaf category. [Allowed Category Paths] (Insert your full list of category paths here, e.g., Technology AI NLP) [Constraints] 1. Output ONLY the category path. No explanation, no preamble. 2. If the text fits multiple paths, output the most specific and relevant one. 3. Use the exact string format provided in the [Allowed Category Paths]. [Input Text] {{text_content}} [Prediction] |
Hierarchical Classification
Hierarchical classification uses relationships among labels to predict categories at multiple levels, from broad groups to fine-grained classes.
See all 89 papers