A Hybrid Transformer Framework For Multi-Label Classification Of Indian Legal Judgments
DOI:
https://doi.org/10.29284/284jtt27Keywords:
legal document classification, multi-label classification, Indian legal judgments, LegalBERT, semi-supervised learning, legal natural language processingAbstract
The rapid growth of digitized judicial records in India has created a need for automatic methods that can organize judgments by legal domain. That requirement is difficult to satisfy because judicial decisions are long, semantically dense, and frequently span more than one area of law. Publicly accessible judgments also contain limited structured metadata, which reduces the usefulness of direct rule-based indexing. This paper presents a hybrid transformer framework for multi-label classification of Indian legal judgments. The proposed system combines LegalBERT-based contextual representations with stacked bidirectional long short-term memory layers so that both local semantics and document-level sequential cues can be captured from long judgments. A two-stage silver-standard annotation strategy is used to construct a scalable training corpus from Supreme Court judgments, while a manually curated gold-standard set is reserved for final evaluation. The framework further incorporates class-weighted binary cross-entropy, label-specific threshold calibration, and pseudo-label guided semi-supervised refinement to improve robustness under label imbalance and limited expert annotation. Experiments on ten legal domains show that the proposed model achieves stronger overall and minority-label performance than sparse-feature, embedding-based, and recurrent neural baselines. Ablation analysis also confirms that pseudo-label refinement, adaptive thresholding, and sequential modeling each contribute to the final gains. The study shows that a carefully designed hybrid architecture, supported by scalable annotation and imbalance-aware inference, provides a practical path for legal document classification in the Indian judicial domain.
References
[1] Y. Yang and X. Liu, "A re-examination of text categorization methods," in Proceedings of the 22nd Annual International ACM SIGIR Conference on Research and Development in Information Retrieval, 1999, pp. 42-49.
[2] G. Tsoumakas and I. Katakis, "Multi-label classification: an overview," International Journal of Data Warehousing and Mining, vol. 3, no. 3, pp. 1-13, 2007.
[3] Z. Yang, D. Yang, C. Dyer, X. He, A. Smola, and E. Hovy, "Hierarchical attention networks for document classification," in Proceedings of NAACL-HLT 2016, 2016, pp. 1480-1489.
[4] I. Chalkidis, M. Fergadiotis, P. Malakasiotis, N. Aletras, and I. Androutsopoulos, "Large-scale multi-label text classification on EU legislation," in Proceedings of ACL 2019, 2019, pp. 6314-6322.
[5] I. Chalkidis, E. Manavopoulos, M. Fergadiotis, and I. Androutsopoulos, "Extreme multi-label legal text classification," in Proceedings of the Natural Legal Language Processing Workshop 2019, 2019, pp. 78-87.
[6] D. Tuggener, P. Von Däniken, C. Peetz, and T. Cieliebak, "LEDGAR: A large-scale multi-label corpus for text classification of legal provisions in contracts," in Proceedings of LREC 2020, 2020, pp. 1235-1241.
[7] I. Chalkidis, M. Fergadiotis, P. Malakasiotis, N. Aletras, and I. Androutsopoulos, "LEGAL-BERT: The Muppets straight out of law school," in Findings of the Association for Computational Linguistics: EMNLP 2020, 2020, pp. 2898-2904.
[8] V. Malik, R. Sanjay, S. Nigam, K. Ghosh, and A. Bhattacharya, "ILDC for CJPE: Indian legal documents corpus for court judgment prediction and explanation," in Proceedings of ACL-IJCNLP 2021, 2021, pp. 4046-4062.
[9] I. Chalkidis, M. Jana, M. Hartung, N. Bommarito, D. Androutsopoulos, D. Katz, and N. Aletras, "LexGLUE: A benchmark dataset for legal language understanding in English," in Proceedings of ACL 2022, 2022, pp. 4310-4330.
[10] A. Kapoor, M. Dhawan, A. Goel, A. T. H. Arjun, A. Bhatnagar, V. Agrawal, A. Agrawal, A. Bhattacharya, P. Kumaraguru, and A. Modi, "HLDC: Hindi legal documents corpus," in Findings of the Association for Computational Linguistics: ACL 2022, 2022, pp. 3521-3536.
[11] I. Nair and N. Modani, "Exploiting language characteristics for legal domain-specific language model pretraining," in Findings of the Association for Computational Linguistics: EACL 2023, 2023.
[12] A. Joshi, S. Paul, A. Sharma, P. Goyal, S. Ghosh, and A. Modi, "IL-TUR: Benchmark for Indian legal text understanding and reasoning," in Proceedings of ACL 2024, 2024, pp. 11460-11476.
[13] J. Devlin, M.-W. Chang, K. Lee, and K. Toutanova, "BERT: Pre-training of deep bidirectional transformers for language understanding," in Proceedings of NAACL-HLT 2019, 2019, pp. 4171-4186.
[14] Q. Le and T. Mikolov, "Distributed representations of sentences and documents," in Proceedings of ICML 2014, 2014, pp. 1188-1196.
[15] K. Nigam, A. McCallum, S. Thrun, and T. Mitchell, "Text classification from labeled and unlabeled documents using EM," Machine Learning, vol. 39, no. 2-3, pp. 103-134, 2000.
[16] I. Beltagy, M. E. Peters, and A. Cohan, "Longformer: The long-document transformer," arXiv preprint arXiv:2004.05150, 2020.
[17] D. D. Lewis, Y. Yang, T. G. Rose, and F. Li, "RCV1: A new benchmark collection for text categorization research," Journal of Machine Learning Research, vol. 5, pp. 361-397, 2004.
Downloads
Published
Issue
Section
License

This work is licensed under a Creative Commons Attribution 4.0 International License.
This work is licensed under a Creative Commons Attribution 4.0 International License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.

