An AI-powered screening tool that analyzes audio clips to detect markers associated with depression. This project uses a Bidirectional Long Short-Term Memory (BiLSTM) neural network trained on the DAIC-WOZ dataset.
- High-Fidelity Audio Analysis: Uses 40 MFCCs plus Delta and Delta-Delta features (120 total features per frame).
- Proactive Pre-processing: Automatic silence trimming and per-sample z-normalization.
- Deep Learning Architecture: Stacked Bidirectional LSTMs with Batch Normalization for temporal feature learning.
- Interactive UI: Clean Streamlit interface for uploading audio and viewing real-time predictions.
- Clone the repository (or navigate to the directory).
- Create and activate a virtual environment:
python -m venv venv # On Windows: venv\Scripts\activate # On Mac/Linux: source venv/bin/activate
- Install dependencies:
pip install streamlit tensorflow librosa numpy pandas scikit-learn
- Train/Download the model:
- Ensure you have
best_depression_model.kerasandmodel_config.jsoninside themodels/folder.
- Ensure you have
- Start the application:
streamlit run app.py
- Analyze: Upload a
.wavfile or use the built-in test audio.
The current model utilizes an optimized threshold of 0.34, achieving:
- Accuracy: ~74%
- Recall (Depressed): ~86%
- F1-Score (Depressed): ~0.76
This tool is for research purposes only and is not a clinical diagnostic instrument. It is designed to demonstrate the potential of vocal biomarkers in mental health screening. Always consult a medical professional for clinical diagnosis.