arXiv:2606.09466cs.CL2026-06中稿 · EMNLP

用自相关掩码让大模型从无标签医学文本中自学分类,提升性能。

DECSELFMASK: Leveraging Unlabeled Text via Self-Relevance-Guided Masking for Decoder-Only Classification

论文配图:DECSELFMASK: Leveraging Unlabeled Text via Self-Relevance-Guided Masking for Decoder-Only Classification
图 1 · 摘自论文原文
  • 通过重要性归因识别文本关键部分,仅掩码这些区域进行自监督训练。
  • 在136个临床任务上平均提升9.1点宏平均F1,优于多种基线方法。
  • 特别适合标注稀缺的医疗领域,可提升小样本分类模型表现。

分类任务依赖标注数据,但获取成本高,尤其在医疗领域,大量数据常缺乏标注。为此,我们提出DecSelfMask(Decoder Self-learning by Masking),一种增强解码器类模型分类性能的方法。该方法基于自学习思想,利用模型从无标签数据生成训练样本,并提出新颖的相关性引导掩码策略:通过重要性归因确定未标注文本中对任务相关的内容,再掩码这些部分,使模型通过下一词预测重建。我们假设此类样本蕴含了无标签数据的结构与语义信息,有助于下游任务。在来自意大利医院的190万条临床笔记中测试136个任务,覆盖5种不同规模和架构的模型。实验显示,该方法在所有模型上均取得稳定增益,相比基础模型提升9.1点宏平均F1,优于持续预训练(+6.3)、合成标签生成(+12.5)及常见基线。结果表明,重要性归因可作为解码器学习的强监督信号,超越现有方法。

原文摘要 · Abstract (English)

Classification tasks require annotated data, which can often be expensive, time-consuming, or even unfeasible to collect. This is the case of the medical domain, where large datasets often have few annotated examples. To address this, we propose DecSelfMask (Decoder Self-learning by Masking), an approach to enhance decoder-only performance on classification tasks. We build on common self-learning approaches by leveraging a model to create training examples from unlabeled data, and propose a novel relevance-guided masking strategy. We use relevance attribution methods to determine what portions of unannotated texts are relevant for a task. We then create self-supervised training examples by masking out those portions, training the model to reconstruct them via next-token-prediction. We hypothesize that those examples convey knowledge about the structure and semantics of unannotated data that can be useful for downstream performance. We test our approach on 136 tasks from a collection of 1.9M clinical notes from an Italian hospital. We quantify DecSelfMask's impact on downstream tasks on 5 models of different scales and families, including a probing analysis. Experiments show consistent gains, outperforming the base models (+9.1 points in Macro F1), continual pretraining (+6.3), synthetic label generation (+12.5), as well as common baselines. The results show that relevance attribution can serve as a powerful source of supervision for decoder learning from unlabeled data, beyond existing methods.

自监督医学文本解码器小样本

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