TRACE: Deployable Tree-Relational Structure Enhancement for Oncology LLMs

Jizheng Lai, Yingyun Li, Ying Qin, Haiyang Qian


Abstract
Large language models are increasingly used in oncology applications, but their predictions are often weakly grounded in explicit medical structure. We present TRACE, a deployable tree-relational enhancement framework for oncology LLMs. TRACE separates expensive offline structure learning from lightweight online inference: oncology concepts and relations are organized into an updatable tree-relational structure, refined using LM-loss-derived evidence, and retrieved at inference time as compact prompt evidence. This design supports task-adaptive evidence selection without requiring supervised labels in the zero-shot setting. Across ten oncology classification tasks and one MedQuAD CancerGov QA benchmark, TRACE improves both label-free evaluation and supervised fine-tuning. Additional analyses show that TRACE improves over vanilla RAG and generic GraphRAG, remains useful under leakage-controlled METABRIC inputs, and produces interpretable evidence paths aligned with clinical reasoning. These results suggest that explicit, updatable medical structure is a practical path toward more accurate and auditable oncology LLM deployment.
Anthology ID:
2026.emnlp-industry.230
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing: Industry Track
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Pepa Atanasova, Gholamreza Haffari, Minghan Wang
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
3836–3853
Language:
URL:
https://aclanthology.org/2026.emnlp-industry.230/
DOI:
Bibkey:
Cite (ACL):
Jizheng Lai, Yingyun Li, Ying Qin, and Haiyang Qian. 2026. TRACE: Deployable Tree-Relational Structure Enhancement for Oncology LLMs. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing: Industry Track, pages 3836–3853, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
TRACE: Deployable Tree-Relational Structure Enhancement for Oncology LLMs (Lai et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-industry.230.pdf