arXiv cs.AI· Animesh Agarwal, Meysam Ghaffari, Nina Fatehi, Anushka Swarup, Carlos Morato·· 13 小时前AI 评分26
从分数到安全集合:用药推荐中的精确安全约束解码
From Scores to Safe Sets: Exact Safety--Constrained Decoding for Medication Recommendation
AI 导读
研究对比了局部修复与精确全局解码在用药推荐中的表现:使用 FLAME 冻结分类器在 MIMIC-III 上的固定药物分数,全局解码器通过混合整数线性规划联合选出得分最高的可行药物集合。
正文 · 原文
Abstract:Recent generative medication recommendation systems ultimately produce sets of drugs, but drug--drug interaction (DDI) constraints depend on the set as a whole. A natural post-hoc approach is to first predict medications and then locally repair interacting pairs by removing one drug at a time. We ask whether this local repair step itself introduces avoidable error. Using fixed medication scores from FLAME's frozen classifier on MIMIC-III, we compare increasingly strong local repair strategies with an exact global decoder while holding the scores, output size, and DDI constraint fixed. The global decoder jointly selects the highest-scoring feasible medication set using mixed-integer linear programming. Across validation and test splits, global decoding consistently achieves higher Jaccard similarity than greedy repair, degree-aware repair, and degree-aware repair strengthened with exhaustive 1-swap local search. Its advantage is small under loose DDI constraints and becomes larger as the constraint tightens; on the test split it reaches +0.0064 Jaccard over the strongest local baseline at the strict DDI operating point. The finding also holds across a fine accuracy--DDI sweep at matched realized DDI exposure and under three score representations. At the strict setting, global decoding retains more true-positive medications while removing more false positives. These results show that medication recommendation is not only a representation-learning problem: even with a frozen clinical model, how its scores are converted into a constrained medication set can materially affect the final prediction.
| Subjects: | Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2610.29228 [cs.AI] |
| (or arXiv:2610.29228v1 [cs.AI] for this version) | |
| https://doi.org/10.48550/arXiv.2610.29228 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Animesh Agarwal [view email]
[v1]
Wed, 21 Oct 2026 05:38:08 UTC (2,191 KB)
来源:arXiv cs.AI · arxiv.org