中文

TD-Suite:面向技术债分类的全能型框架

软件工程 2025-04-16 v1 机器学习

摘要

鉴于技术债是需要精细管理工具来应对的持续性和重大挑战,TD-Suite 提供了一个综合的软件框架,专为自动化完成软件项目中技术债分类这一复杂任务而设计。它利用了来自 Transformer 模型的先进自然语言理解能力,用于分析文本性质的产出物,如开发人员在问题报告中的讨论,其中隐藏着技术债的细微指示。TD-Suite 提供了从端到端的无缝管道,涵盖初始数据摄取、严格的预处理、模型训练、深入评估以及最终推断。这使其能够支持简单的二进制分类(技术债或无技术债),以及更有价值的识别具体类别(如代码债、设计债或文档债),从而实现更有针对性的管理策略。为确保生成的模型在现实世界、常不平衡的数据集上具有鲁棒性和可靠性能力,TD-Suite 集成了关键训练方法:K 折交叉验证评估泛化能力,早停机制防止过拟合训练数据,类别加权策略有效解决数据分布不平衡的问题。除了核心功能,鉴于可持续性日益重要,框架还集成了跟踪和报告与模型训练过程相关的碳排放。它还 features 一个用户友好的 Gradio Web 界面,采用 Docker 容器设置,简化了模型的交互、评估和推断。

关键词

引用

@article{arxiv.2504.11085,
  title  = {TD-Suite: All Batteries Included Framework for Technical Debt Classification},
  author = {Karthik Shivashankar and Antonio Martini},
  journal= {arXiv preprint arXiv:2504.11085},
  year   = {2025}
}

备注

In submission