Integrating Lightweight Transformers for Cross-Project Bug Severity Classification: An Applied AI Approach in Software Engineering
Source abstract
Abstract
Bug severity classification is an important task in software maintenance because it supports bug triage and resource allocation. However, newly created or evolving projects often lack sufficient labeled data, making cross-project severity prediction challenging due to domain shift and class imbalance. In this paper, we investigate cross-project bug severity classification using lightweight transformer models under practical deployment constraints. Specifically, DistilBERT and TinyBERT are employed and evaluated within a unified cross-project learning framework. Experiments are conducted on large-scale Mozilla bug repositories under both single-source and multi-source transfer settings. Macro-averaged F1 is used as the primary evaluation metric to ensure balanced assessment across severity levels. The results indicate that cross-project performance is strongly influenced by source–target pairing, reflecting the impact of domain shift. Multi-source training generally improves performance across several transfer scenarios, particularly for minority severity classes, although the improvements remain moderate. DistilBERT achieves higher overall performance, whereas TinyBERT shows comparable trends with only a small reduction in Macro-F1, suggesting a favorable trade-off between predictive performance and model efficiency. These findings suggest that lightweight transformer models can support practical bug triage processes by providing relatively consistent and computationally efficient severity predictions across projects, particularly in environments with limited labeled data and computational resources.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Software Engineering Research · Software Engineering Techniques and Practices · Software System Performance and Reliability
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Department of Computer Science, Faculty of Informatics, Mahasarakham University, Kantarawichai District, Maha Sarakham 44150, Thailand
Funding
Mahasarakham University
References (36)
Show source-provided references
1. Arokiam, J., and Bradbury, J.S. (2020). Automatically Predicting Bug Severity Early in the Development Process. Proceedings of the 42nd International Conference on Software Engineering: New Ideas and Emerging Results, Seoul, Republic of Korea, IEEE.10.1145/3377816.3381738 ↗
2. Towards More Accurate Severity Prediction and Fixer Recommendation of Software Bugs10.1016/j.jss.2016.02.034 ↗
3. Deep Mining of Open Source Software Bug Repositories
4. Jarzabek, S., Poniszewska-Marańda, A., and Madeyski, L. (2020). Semantic Categorization of Software Bug Repositories for Severity Assignment Automation. Integrating Research and Practice in Software Engineering, Springer.10.1007/978-3-030-26574-8 ↗
5. Text Mining Approaches for Dependent Bug Report Assembly and Severity Prediction
6. Sarawan, K., Polpinij, J., and Luaphol, B. (2023). Machine Learning-Based Methods for Identifying Bug Severity Level from Bug Reports. Proceedings of the International Conference on Computing and Information Technology, Bangkok, Thailand, Springer.10.1007/978-3-031-30474-3_17 ↗
7. A Systematic Literature Review of Software Defect Prediction using Deep Learning10.3844/jcssp.2021.490.510 ↗
8. Learning Software Bug Reports: A Systematic Literature Review
9. SevPredict: Exploring the Potential of Large Language Models in Software Maintenance
10. Rumman, M., Roy, E., Zaman, A., and Bradbury, J.S. (2025). A Contrastive Learning Approach to Bug Severity Classification with Large Language Model Embeddings. Proceedings of the 49th Annual Computers, Software, and Applications Conference (COMPSAC), Toronto, ON, Canada, IEEE Computer Society.10.1109/COMPSAC65507.2025.00172 ↗
11. Zimmermann, T., Nagappan, N., Gall, H., Giger, G., and Murphy, B. (2009). Cross-Project Defect Prediction: A Large-Scale Experiment on Data vs. Domain vs. Process. Proceedings of the Joint European Software Engineering Conference and Symposium on the Foundations of Software Engineering, Singapore, Association for Computing Machinery.10.1145/1595696.1595713 ↗
12. Issue Classification with LLMs: An Empirical Study of the NASA flight Software Systems10.1016/j.jss.2026.112851 ↗
13. Developing bug severity prediction models using word2vec
14. Cross-Project Bug Type Prediction Based on Transfer Learning10.1007/s11219-019-09467-0 ↗
15. Nam, J., Pan, S.J., and Kim, S. (2013). Transfer Defect Learning. Proceedings of the International Conference on Software Engineering, San Francisco, CA, USA, Association for Computing Machinery.10.1109/ICSE.2013.6606584 ↗
16. A Survey on Transfer Learning for Cross-Project Defect Prediction10.1109/ACCESS.2024.3424311 ↗
17. Improving Automated Program Repair with Domain Adaptation
18. Li, Z., Li, Y., Li, T., Du, M., Wu, B., Cao, Y., Xie, X., Li, Y., and Liu, Y. (2024, January 20–25). Unveiling Project-Specific Bias in Neural Code Models. Proceedings of the Joint International Conference on Computational Linguistics, Language Resources and Evaluation, Torino, Italy.10.63317/3q3jx6p533c9 ↗
19. Mock, M., Forrer, T., and Russo, B. (2025, January 1–3). Cross-Domain Evaluation of Transformer-Based Vulnerability Detection on Open and Industry Data. Proceedings of the 26th International Conference on Product-Focused Software Process Improvement, Salerno, Italy.10.1007/978-3-032-12092-2_3 ↗
20. Improving Bug Severity Prediction with Domain-Specific Representation Learning10.1109/ACCESS.2023.3279205 ↗
21. A Weighted Multi-Source Domain Adaptation Approach for Surface Defect Detection10.1049/ipr2.12484 ↗
22. Promises and Perils of Using Transformer-Based Models for SE Research10.1016/j.neunet.2024.107067 ↗
23. On the Validity of Pre-Trained Transformers for Natural Language Processing in the Software Engineering Domain10.1109/TSE.2022.3178469 ↗
24. Keys4BR: Key Sentences-based Model Fine-Tuning for Better Semantic Representation of Bug Reports10.1016/j.infsof.2025.107943 ↗
25. Stankevicius, L., and Lukoševičius, M. (2024). Extracting Sentence Embeddings from Pretrained Transformer Models. Appl. Sci., 14.10.3390/app14198887 ↗
26. Grishina, A., Hort, M., and Moonen, L. (2023). The EarlyBIRD Catches the Bug: On Exploiting Early Layers of Encoder Models for More Efficient Code Classification. Proceedings of the 31st ACM Joint European Software Engineering Conference and Symposium on the Foundations of Software Engineering, San Francisco, CA, USA, Association for Computing Machinery.10.1145/3611643.3616304 ↗
27. Sanh, V., Debut, L., Chaumond, J., and Wolf, T. (2019, January 13). DistilBERT, a Distilled Version of BERT: Smaller, Faster, Cheaper and Lighter. Proceedings of the 5th Workshop on Energy Efficient Machine Learning and Cognitive Computing (EMC2-NIPS), Vancouver, BC, Canada.
28. Jiao, X., Yin, Y., Shang, L., Jiang, X., Chen, X., Li, L., Wang, F., and Liu, Q. (2020, January 16–20). TinyBERT: Distilling BERT for Natural Language Understanding. Proceedings of the Findings of the Association for Computational Linguistics: EMNLP, Online.10.18653/v1/2020.findings-emnlp.372 ↗
29. Domain Adaptation via Transfer Component Analysis10.1109/TNN.2010.2091281 ↗
30. Unsupervised Domain Adaptation via Discriminative Manifold Propagation and Maximum Mean Discrepancy
31. Chen, C., Chen, Z., Jiang, B., and Jin, X. (2019). Joint Domain Alignment and Discriminative Feature Learning for Unsupervised Deep Domain Adaptation. Proceedings of the Thirty-Third AAAI Conference on Artificial Intelligence and Thirty-First Innovative Applications of Artificial Intelligence Conference and Ninth AAAI Symposium on Educational Advances in Artificial Intelligence, Honolulu, HI, USA, AAAI Press.10.1609/aaai.v33i01.33013296 ↗
32. Balanced Adversarial Tight Matching for Cross-Project Defect Prediction10.1049/2024/1561351 ↗
33. Zhao, H., Zhang, S., Wu, G., Costeira, J.P., Moura, J.M.F., and Gordon, G.J. (2018). Adversarial Multiple Source Domain Adaptation. Proceedings of the 32nd Conference on Neural Information Processing Systems (NeurIPS), Montréal, QC, Canada, Curran Associates Inc.
34. Li, J., Xu, Z., Yongkang, W., Zhao, Q., and Kankanhalli, M. (2020, January 6–10). GradMix: Multi-Source Transfer across Domains and Tasks. Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV), Tucson, Arizona.10.1109/WACV45572.2020.9093343 ↗
35. Lee, M.C.H., Braet, J., and Springael, J. (2024). Performance Metrics for Multilabel Emotion Classification: Comparing Micro, Macro, and Weighted F1-Scores. Appl. Sci., 14.10.3390/app14219863 ↗
36. A Closer Look at Classification Evaluation Metrics and a Critical Reflection of Common Evaluation Practice10.1162/tacl_a_00675 ↗