arXiv:2509.02072cs.LGcs.IR2025-09

用精简抽象+对抗训练,解决工伤报告分类中的数据少与不平衡问题。

Abex-rat: Synergizing Abstractive Augmentation and Adversarial Training for Classification of Occupational Accident Reports

  • 先用提示引导的LLM提炼关键语义生成摘要,再扩展成多样合成数据。
  • 在OSHA数据集上达到90.32%的宏平均F1,超越传统方法和大模型微调。
  • 轻量级设计适合资源受限的垂直领域分类任务,代码将开源。

职业事故报告的自动分类对工作场所安全分析至关重要,但长期受制于严重的类别不平衡和数据稀缺。本文提出ABEX-RAT,一种资源高效的框架,融合生成式数据增强与鲁棒对抗学习。不同于计算昂贵的大语言模型(LLMs)微调,本方法采用两阶段抽象扩展(ABEX)流程:首先利用提示引导的LLM将标签相关语义提炼为简洁摘要,再将其扩展为多样化合成样本以平衡数据分布;随后采用随机对抗训练(RAT)协议训练轻量级分类器,通过随机注入扰动提升泛化能力,且计算开销极低。在OSHA数据集上的实验表明,ABEX-RAT达到新的最佳性能,宏平均F1达90.32%,显著优于传统基线及微调的大模型。结果证实,针对性增强结合鲁棒训练是专业领域分类中更优、更高效的数据解决方案。源代码将在论文录用后公开。

原文摘要 · Abstract (English)

The automatic classification of occupational accident reports is pivotal for workplace safety analysis but is persistently hindered by severe class imbalance and data scarcity. In this paper, we propose ABEX-RAT, a resource-efficient framework that synergizes generative data augmentation with robust adversarial learning. Unlike computationally expensive large language models (LLMs) fine-tuning, our approach employs a two-stage abstractive-expansive (ABEX) pipeline: it first utilizes a prompt-guided LLM to distill label-critical semantics into concise abstracts, which are then expanded into diverse synthetic samples to balance the data distribution. Subsequently, we train a lightweight classifier using a random adversarial training (RAT) protocol, which stochastically injects perturbations to enhance generalization without significant computational overhead. Experimental results on the OSHA dataset demonstrate that ABEXRAT establishes a new state-of-the-art, achieving a Macro-F1 score of 90.32% and significantly outperforming both traditional baselines and fine-tuned large models. This confirms that targeted augmentation combined with robust training offers a superior, data-efficient alternative for specialized domain classification. The source code will be made publicly available upon acceptance.

分类数据增强对抗训练垂直领域

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。