LLM Reasoning 相关度: 7/10

MADE: A Living Benchmark for Multi-Label Text Classification with Uncertainty Quantification of Medical Device Adverse Events

Raunak Agarwal, Markus Wenzel, Simon Baur, Jonas Zimmer, George Harvey, Jackie Ma
arXiv: 2604.15203v1 发布: 2026-04-16 更新: 2026-04-16

AI 摘要

提出了MADE,一个医疗设备不良事件的多标签文本分类benchmark,关注不确定性量化。

主要贡献

  • 提出了MADE benchmark,避免数据污染
  • 评估了多种模型在多标签文本分类上的性能和不确定性量化
  • 分析了不同模型和不确定性量化方法的优劣

方法论

构建基于医疗设备不良事件报告的多标签数据集,采用时序分割,评估多种模型并进行不确定性量化。

原文摘要

Machine learning in high-stakes domains such as healthcare requires not only strong predictive performance but also reliable uncertainty quantification (UQ) to support human oversight. Multi-label text classification (MLTC) is a central task in this domain, yet remains challenging due to label imbalances, dependencies, and combinatorial complexity. Existing MLTC benchmarks are increasingly saturated and may be affected by training data contamination, making it difficult to distinguish genuine reasoning capabilities from memorization. We introduce MADE, a living MLTC benchmark derived from {m}edical device {ad}verse {e}vent reports and continuously updated with newly published reports to prevent contamination. MADE features a long-tailed distribution of hierarchical labels and enables reproducible evaluation with strict temporal splits. We establish baselines across more than 20 encoder- and decoder-only models under fine-tuning and few-shot settings (instruction-tuned/reasoning variants, local/API-accessible). We systematically assess entropy-/consistency-based and self-verbalized UQ methods. Results show clear trade-offs: smaller discriminatively fine-tuned decoders achieve the strongest head-to-tail accuracy while maintaining competitive UQ; generative fine-tuning delivers the most reliable UQ; large reasoning models improve performance on rare labels yet exhibit surprisingly weak UQ; and self-verbalized confidence is not a reliable proxy for uncertainty. Our work is publicly available at https://hhi.fraunhofer.de/aml-demonstrator/made-benchmark.

标签

Multi-label Text Classification Uncertainty Quantification Medical Device Adverse Events Benchmark

arXiv 分类

cs.CL