fdabbiras commited on
Commit
fb90326
·
verified ·
1 Parent(s): 176de31

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -75
README.md CHANGED
@@ -1,76 +1,10 @@
1
- # Explainable Multi-Layer Dynamic Ensemble Framework Optimized for Depression Detection and Severity Prediction
2
-
3
- <p align="justify">
4
- Depression is a common mental health issue, especially among older adults, and early detection is critical to effective intervention and treatment. This study proposes an explainable multilayer dynamic ensemble framework optimized for detecting depression and predicting its severity. Using data from real individuals from the National Social Life, Health, and Aging Project (NSHAP), the framework integrates classical machine learning models, static ensemble techniques, and dynamic ensemble selection (DES) approaches to build a two-stage framework. The depression detection process is performed in the first stage, and the depression severity is predicted in the second stage only for depressed patients. Among the models evaluated, the FIRE-KNOP DES algorithm exhibited the highest performance, achieving 88.33\% accuracy in detecting depression and 83.68\% accuracy in predicting depression severity. A key innovation of this study is the incorporation of explainable AI (XAI) techniques to enhance model interpretability, making the framework more suitable for clinical applications. We explore different global and local XAI features. Model explainability highlights the significance of medically relevant mental and non-mental health features in improving the model's performance, enhancing the domain experts' trustworthiness in the model decisions. These findings offer a promising foundation for future applications of dynamic ensemble frameworks in mental health assessments.
5
- </p>
6
-
7
- For more details, please check (link here)
8
-
9
  ---
10
-
11
- ## The Architecture of the Proposed Framework
12
- <div align="center">
13
- <img src="https://github.com/InfoLab-SKKU/DES4Depression/blob/main/Proposed%20Architecture_page-0001.jpg" alt="Proposed Architecture" width="500"/>
14
- </div>
15
-
16
- ---
17
-
18
- ## Guideline for Reproducing Results
19
-
20
- ### Obtaining and Processing Data
21
-
22
- 1. The data can be found on [NSHAP - ICPSR20541](https://doi.org/10.3886/ICPSR20541.v10) under the **Data & Documentation** tab. Download **DS1 Core Data, Public-Use**.
23
-
24
- 2. Run the following lines in R to create a CSV file:
25
-
26
- ```r
27
- # Load the .rda file and export it to CSV
28
- my_data_frame = load("./20541-0001-Data.rda")
29
- ls()
30
- write.csv(my_data_frame, file = "data.csv", row.names = FALSE)
31
- ```
32
-
33
- 3. Data can be processed in `dataPreprocessing.ipynb` in the code folder. Run the code sequentially. At the end, two outputs will be generated:
34
- - **3labelv4Classification.csv** for Detection and Severity Prediction tasks.
35
- - **3labelv4Regression.csv** for Scale Prediction task.
36
-
37
- ### Detection and Severity Prediction Task
38
-
39
- - **Metrics Results**:
40
-
41
- The `detectionLayer.ipynb` and `severityPredictionLayer.ipynb` files in the code folder follow similar experimental methods. The code needs to be run sequentially from **Start** up to **Training (classic/static)**. At this point, you may choose to:
42
-
43
- 1. **Train with classical/static classifiers**.
44
-
45
- - To train with classic models, uncomment the models you want to use in the **Classic Classifiers** code cell and run it.
46
- - To train with static models, uncomment the models you want to use in the **Static Classifiers** code cell and run it.
47
- - To get the average over 10 states, ROC curves, and FN curve, run the **Post Training (classic/static)** code cells only after running either the **Classic Classifiers** or **Static Classifiers** code cell.
48
- - Below the **Post Training (classic/static)** code cells, you'll find hyperparameter optimization cells.
49
-
50
- 2. **Train with DES methods**.
51
-
52
- - Go immediately to **DES Training (all)**, uncomment the base classifiers that you want to use for DES, and run the code cells.
53
-
54
- - **SHAP Results**:
55
-
56
- - SHAP results can be generated in the **Shap (will mostly be exported files)** section, which is located towards the bottom of the `.ipynb` file.
57
-
58
- ### Scale Prediction Task
59
-
60
- - **Metrics Results**:
61
-
62
- - The `regressionLayer.ipynb` file in the code folder can be used for the Scale Prediction task.
63
- - Run the code sequentially from **Start** up to before **Regression Training**.
64
- - Then, run **Regression Training** to get the metrics.
65
-
66
- - Hyperparameter optimization can be found towards the bottom of the `.ipynb` file.
67
-
68
- - **SHAP and FN**:
69
-
70
- - Run the code in the **SHAP and FN** sections to generate the FN diagram and SHAP files.
71
-
72
- ---
73
-
74
- ## Citation
75
-
76
- We would appreciate it if you consider citing our work when using our code.
 
 
 
 
 
 
 
 
 
1
  ---
2
+ title: Judul Aplikasi Kamu
3
+ emoji: 🤖
4
+ colorFrom: blue
5
+ colorTo: red
6
+ sdk: streamlit # Ganti dengan 'gradio', 'streamlit', atau 'docker'
7
+ sdk_version: 1.25.0 # (Opsional) Hapus baris ini jika ingin versi terbaru otomatis
8
+ app_file: app.py # Pastikan nama file python utamamu adalah app.py
9
+ pinned: false
10
+ ---