用强化学习动态决定检测哪些基因组数据,省钱又准。
SDM-Q: Cost-Aware Staged Decision-Making for Multi-Omics Classification with Deep Q-Learning

- 将多组学诊断转为分阶段决策问题,按需选择检测项目。
- 在乳腺癌和肾癌数据中99%、95%仅靠一种数据就准确分类。
- 适合临床场景中资源有限但需精准诊断的研究者。
多组学数据为疾病表型提供互补的分子特征,在精准医疗中对疾病诊断和亚型分类至关重要。然而,获取完整的多组学谱图成本高昂且耗时,而现有深度学习方法通常假设推理时所有组学数据均可用,导致大量冗余,实用性受限。为此,我们提出SDM-Q,一种基于深度Q学习的自适应、成本感知多组学分类框架。将多组学诊断重新建模为有限时域的序贯决策问题,当前已获取的组学数据构成每个阶段的诊断状态,动作-价值函数决定是否继续采集新组学数据或终止决策并输出预测结果。奖励仅在终态定义,由分类正确性与累计数据获取成本共同决定。引入逆向阶段优化策略以提升策略一致性和训练稳定性。在四个公开多组学数据集(ROSMAP、LGG、BRCA、KIPAN)上的实验表明,SDM-Q有效减少了冗余的数据采集,同时保持与使用完整组学输入方法相当的分类性能。在BRCA和KIPAN数据集中,分别有超过99%和95%的样本仅通过单一组学模态即实现准确分类,而ROSMAP和LGG数据集上平均采集模态数低于2个。结果表明,成本感知的序贯决策为提升精准医疗工作流效率提供了有效范式。
原文摘要 · Abstract (English)
Multi-omics data provide complementary molecular characterizations of disease phenotypes and play an important role in disease diagnosis and subtype classification in precision medicine. However, acquiring complete multi-omics profiles is expensive and time-consuming, while most existing deep learning methods assume full modality availability during inference, resulting in substantial redundancy and limited practicality in clinical settings. To address this issue, we propose SDM-Q, a reinforcement learning framework for adaptive and cost-aware multi-omics classification. Specifically, multi-omics diagnosis is reformulated as a finite-horizon sequential decision problem, where the currently acquired omics modalities define the diagnostic state at each stage. An action--value function determines whether to acquire an additional modality or terminate the decision process and output the final prediction. To balance diagnostic utility and acquisition cost, the reward is defined only at the terminal stage and jointly determined by classification correctness and cumulative modality acquisition cost. A backward stage-wise optimization strategy is introduced to improve policy consistency and training stability. Experiments on four public multi-omics datasets, including ROSMAP, LGG, BRCA, and KIPAN, demonstrate that SDM-Q effectively reduces redundant modality acquisition while maintaining competitive classification performance compared with methods using complete multi-omics inputs. In the BRCA and KIPAN datasets, more than 99\% and 95\% of subjects, respectively, achieve accurate classification using only a single omics modality, while the average number of acquired modalities remains below two for ROSMAP and LGG. These results suggest that cost-aware sequential decision-making provides an effective paradigm for improving the efficiency of precision medicine workflows.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。