MAIT为医疗数据提供可解释的机器学习工具,支持分类、生存分析与回归。
Medical artificial intelligence toolbox (MAIT): an explainable machine learning framework for binary classification, survival modelling, and regression analyses
- 统一框架处理高维、缺失、不平衡等医疗数据难题
- 支持特征重要性分析与概率阈值优化,提升模型可解释性
- 适合医学研究者快速上手,也支持专家定制开发
机器学习虽能应对多样化的医学研究问题,但缺乏整合性框架制约其应用。为此,我们推出医学人工智能工具箱(MAIT),一个开源的Python流程,用于在表格数据上构建与评估二分类、回归及生存模型。MAIT解决高维、类别不平衡、变量类型混合和缺失值等关键挑战,并符合TRIPOD+AI报告规范,增强透明度。面向初学者提供自动化配置,面向专家提供可定制代码,支持两类核心场景:发现(通过统一评分如SHAP进行特征重要性分析)与预测(优化模型开发与部署)。此外,MAIT提出新方法,包括二分类概率阈值微调、累积风险曲线转二分类、混合数据类型可视化增强,以及通过半监督学习处理删失数据,以适应多种数据约束与研究设计。我们在GitHub提供四个公开数据集的详细教程,展示如何用MAIT提升医学研究中机器学习模型的实现与解释能力。
原文摘要 · Abstract (English)
While machine learning offers diverse techniques suitable for exploring various medical research questions, a cohesive synergistic framework can facilitate the integration and understanding of new approaches within unified model development and interpretation. We therefore introduce the Medical Artificial Intelligence Toolbox (MAIT), an explainable, open-source Python pipeline for developing and evaluating binary classification, regression, and survival models on tabular datasets. MAIT addresses key challenges (e.g., high dimensionality, class imbalance, mixed variable types, and missingness) while promoting transparency in reporting (TRIPOD+AI compliant). Offering automated configurations for beginners and customizable source code for experts, MAIT streamlines two primary use cases: Discovery (feature importance via unified scoring, e.g., SHapley Additive exPlanations - SHAP) and Prediction (model development and deployment with optimized solutions). Moreover, MAIT proposes new techniques including fine-tuning of probability threshold in binary classification, translation of cumulative hazard curves to binary classification, enhanced visualizations for model interpretation for mixed data types, and handling censoring through semi-supervised learning, to adapt to a wide set of data constraints and study designs. We provide detailed tutorials on GitHub, using four open-access data sets, to demonstrate how MAIT can be used to improve implementation and interpretation of ML models in medical research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。