Enhancing Text Classification with a Novel Multi-Agent Collaboration Framework Leveraging BERT
Hediyeh Baban, Sai A Pidapar, Aashutosh Nema, Sichen Lu

TL;DR
This paper presents a multi-agent collaboration framework that enhances BERT-based text classification accuracy and robustness by integrating specialized agents for analysis and consensus, achieving significant performance improvements.
Contribution
Introduces a novel multi-agent system that collaborates with BERT to improve text classification accuracy and robustness across diverse tasks.
Findings
Achieves a 5.5% increase in accuracy over standard BERT classifiers.
Demonstrates effectiveness across multiple benchmark datasets.
Advances multi-agent collaboration in natural language processing.
Abstract
We introduce a novel multi-agent collaboration framework designed to enhance the accuracy and robustness of text classification models. Leveraging BERT as the primary classifier, our framework dynamically escalates low-confidence predictions to a specialized multi-agent system comprising Lexical, Contextual, Logic, Consensus, and Explainability agents. This collaborative approach allows for comprehensive analysis and consensus-driven decision-making, significantly improving classification performance across diverse text classification tasks. Empirical evaluations on benchmark datasets demonstrate that our framework achieves a 5.5% increase in accuracy compared to standard BERT-based classifiers, underscoring its effectiveness and academic novelty in advancing multi-agent systems within natural language processing.
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Taxonomy
TopicsTopic Modeling · Explainable Artificial Intelligence (XAI) · Sentiment Analysis and Opinion Mining
MethodsRefunds@Expedia|||How do I get a full refund from Expedia? · Attention Is All You Need · Linear Layer · Attention Dropout · Softmax · WordPiece · Weight Decay · Multi-Head Attention · Dropout · Residual Connection
