Search papers, labs, and topics across Lattice.
This paper introduces DiagLoop, a novel counterfactual data flywheel designed to enhance diagnostic capabilities in large language models (LLMs) by generating training supervision from codified clinical guidelines and physical relations. The method employs a training-only teacher to create counterfactual scenarios, while a hybrid checker ensures the validity of these scenarios, enabling the model to learn through symptom abstraction and causal reasoning. The resulting 8B model demonstrates significant improvements in path correctness, outperforming conventional baselines and proprietary references across multiple industrial systems and disease categories, achieving gains of up to 11.6 points.
Counterfactual training can boost diagnostic accuracy in LLMs by over 11 points without relying on expert-annotated cases.
Causal diagnostic models must explain how conclusions follow from evidence because diagnoses guide repairs and treatments. Yet serious cases are scarce, records rarely contain reasoning paths, and data transfer poorly across configurations, complicating local deployment. We present DiagLoop, a counterfactual data flywheel that converts codified physical relations or clinical guidelines, authored once per mechanism family, into training supervision beyond recorded cases. A training-only teacher proposes counterfactual worlds by varying causes, contexts, and observations, while an independent hybrid checker admits only valid worlds. The student reasons through symptom abstraction, causal-chain construction, and root-cause attribution. Stage-specific criteria identify its earliest failure. For nonterminal failures, a bounded repair probes downstream competence, and the resulting weakness profile guides subsequent data generation. Stage-localized reinforcement learning updates only the model-generated continuation, while replay and preservation reduce forgetting. The same criteria govern admission, attribution, reward, and regeneration through checks separate from the proposer. Using only synthesized scenarios and no case-level expert reasoning annotations, the resulting 8B model improves strict path correctness over the strongest conventional baseline. Gains are 11.6 points across eight industrial systems and 5.5 points across ten disease categories. Gains over a deranged-routing control are 3.9 and 2.3 points, respectively. The model also exceeds the evaluated proprietary references in both domains, even when they receive few-shot examples or the specification in context.
