Learning Optimal Biomarker-Guided Treatment Policy for Chronic Disorders

Bin Yang,Xingche Guo,Ji Meng Loh,Qinxia Wang,Yuanjia Wang
DOI: https://doi.org/10.48550/arXiv.2305.13852
2023-05-23
Abstract:Electroencephalogram (EEG) provides noninvasive measures of brain activity and is found to be valuable for diagnosis of some chronic disorders. Specifically, pre-treatment EEG signals in alpha and theta frequency bands have demonstrated some association with anti-depressant response, which is well-known to have low response rate. We aim to design an integrated pipeline that improves the response rate of major depressive disorder patients by developing an individualized treatment policy guided by the resting state pre-treatment EEG recordings and other treatment effects modifiers. We first design an innovative automatic site-specific EEG preprocessing pipeline to extract features that possess stronger signals compared with raw data. We then estimate the conditional average treatment effect using causal forests, and use a doubly robust technique to improve the efficiency in the estimation of the average treatment effect. We present evidence of heterogeneity in the treatment effect and the modifying power of EEG features as well as a significant average treatment effect, a result that cannot be obtained by conventional methods. Finally, we employ an efficient policy learning algorithm to learn an optimal depth-2 treatment assignment decision tree and compare its performance with Q-Learning and outcome-weighted learning via simulation studies and an application to a large multi-site, double-blind randomized controlled clinical trial, EMBARC.
Applications
What problem does this paper attempt to address?
This paper aims to solve the problem of how to design individualized treatment strategies by using resting - state pre - processed electroencephalogram (EEG) recordings and other treatment - effect modifiers to improve the treatment response rate of patients with major depressive disorder (MDD). Specifically, the paper focuses on how to use specific features in EEG signals (such as activities in the alpha - wave and theta - wave frequency bands) to guide the selection of antidepressant treatments, because these signals are related to the response to antidepressant drugs, and the response rate of antidepressant treatments is usually low. To achieve this goal, the paper proposes a comprehensive process, including: 1. **EEG Pre - processing and Feature Extraction**: An innovative automatic site - specific EEG pre - processing pipeline is designed to extract features with stronger signals from the raw data. This step is crucial for reducing noise and standardizing data from different acquisition points. 2. **Conditional Average Treatment Effect Estimation**: The causal forest method is used to estimate the conditional average treatment effect, and the double - robust technique is adopted to improve the estimation efficiency. Through this method, researchers can provide evidence of heterogeneous treatment effects, which is a result that cannot be obtained by traditional methods. 3. **Learning of Optimal Treatment Allocation Decision Trees**: An efficient policy - learning algorithm is used to learn an optimal treatment - allocation decision tree with a depth of 2, and its performance is compared with Q - Learning and outcome - weighted learning methods through simulation studies and applications to the EMBARC multi - center, double - blind, randomized controlled clinical trial. In conclusion, this paper attempts to optimize the treatment selection for MDD patients by integrating EEG features and other treatment - effect modifiers, thereby improving the effectiveness and personalization level of treatment.