ADHAM: Additive Deep Hazard Analysis Mixtures for Interpretable Survival Regression

2509.07108v1 stat.ML, cs.LG 2025-09-11
Авторы:

Mert Ketenci, Vincent Jeanselme, Harry Reyes Nieva, Shalmali Joshi, Noémie Elhadad

Резюме на русском

## Контекст Сбои сердечно-сосудистой системы и другие тяжелые заболевания часто требуют моделирования временных интервалов до событий. Традиционные модели временных интервалов, такие как Cox proportional hazards и parametric models, не всегда предоставляют достаточную точность или интерпретируемость для клинических приложений. Нейронные сети, в свою очередь, предлагают высокую точность, но часто являются неинтерпретируемыми. Это недостаток значительно ограничивает их применение в клинической практике, где важно понимать, как отдельные факторы (например, родительские заболевания) влияют на временные интервалы до событий. Для решения этой проблемы мы предлагаем **ADHAM**, интерпретируемую нейронную сеть для моделирования временных интервалов до событий. ## Метод **ADHAM** — это нейронная сеть с аддитивным структурированием, которая моделирует временные интервалы до событий с помощью комбинации функций риска, которые зависят от признаков. Мы предлагаем метод, который позволяет определить группы подобных объектов с помощью нейронных сетей, а затем оценить их влияние на временные интервалы. Для того, чтобы сократить число групп до наиболее значимых, мы вводим пост-тренировочный процесс, который объединяет похожие группы на основе их характеристик. ## Результаты Мы провели эксперименты с **ADHAM** на нескольких реальных датасетах, включая данные о раке и заболеваниях сердца. Эти эксперименты показали, что наша модель предоставляет интерпретируемые результаты, позволяющие понять, какие факторы влияют на временные интервалы. Например, в моделировании временных интервалов до рака мы можем увидеть, что группы людей с определенными генетическими факторами имеют более высокий риск. Эти результаты показали, что **ADHAM** применима как для групповых, так и для индивидуальных прогнозов. ## Значимость Мы видим применение нашей модели в санитарной статистике, клинических решениях и предложении новых лекарств. ADHAM позволяет упростить понимание влияния разных факторов на временные интервалы, что может помочь врачам принимать более обоснованные решения. Модель характеризуется высокой точностью и интерпретируемостью, что делает ее уникальной в сравнении с другими моделями, которые могут предложить только точную, но неинтерпретируемую, оценку. ## Выводы **ADHAM** — это новая, интерпретируемая модель для моделирования временных интервалов до событий. Мы продемонстрировали ее высокую точность и интерпретируемость на реальных данных. У нас есть планы продолжать развивать **ADHAM**, сделав ее еще более точной и применимой к различным клиническим ситуациям. Мы также планируем провести дополнительные

Abstract

Survival analysis is a fundamental tool for modeling time-to-event outcomes in healthcare. Recent advances have introduced flexible neural network approaches for improved predictive performance. However, most of these models do not provide interpretable insights into the association between exposures and the modeled outcomes, a critical requirement for decision-making in clinical practice. To address this limitation, we propose Additive Deep Hazard Analysis Mixtures (ADHAM), an interpretable additive survival model. ADHAM assumes a conditional latent structure that defines subgroups, each characterized by a combination of covariate-specific hazard functions. To select the number of subgroups, we introduce a post-training refinement that reduces the number of equivalent latent subgroups by merging similar groups. We perform comprehensive studies to demonstrate ADHAM's interpretability at the population, subgroup, and individual levels. Extensive experiments on real-world datasets show that ADHAM provides novel insights into the association between exposures and outcomes. Further, ADHAM remains on par with existing state-of-the-art survival baselines in terms of predictive performance, offering a scalable and interpretable approach to time-to-event prediction in healthcare.

Ссылки и действия