ResearchDaffodil International UniversityWork period · 2023-2024Published Research

Mental Health Diagnosis Using CNNs and Vision Transformers

CNN–Vision Transformer research for binary classification of Bengali voice samples, published in Journal of Voice.

Journal article · 2024
Mental Health Diagnosis Using CNNs and Vision Transformers project workflow and system architecture
Project workflow and system architecture overview. Open full-size diagram

Workflow at a glance

  1. Prepare voice data

    Represent Bengali recordings as spectrograms.

  2. Learn CNN + ViT features

    Study local and broader acoustic patterns.

  3. Classify study labels

    Predict the binary stable/unstable labels.

  4. Evaluate research results

    Assess study metrics and interpretation.

Project facts & reported results

~91%
accuracy

Reported result

Reported study result; see linked paper for evaluation protocol.

~0.97
ROC-AUC

Reported result

Reported study result; see linked paper for evaluation protocol.

85 Bengali voice recordings
dataset

Implementation fact

Project Overview

Problem Statement

Small voice datasets and acoustic variation make mental-health-related classification difficult to evaluate reliably. This study explores spectrogram learning in the Bengali voice domain.

Approach & Methodology

Converted audio to log-mel spectrograms, treating voice analysis as a vision-driven learning problem. Proposed hybrid CNN-ViT architecture: CNN layers extract local acoustic patterns, ViT captures long-range dependencies in temporal sequences. Applied class imbalance handling (SMOTE), data augmentation, and robust evaluation methodology. Used ethically collected Bengali voice data from mental health institutions in Bangladesh.

Outcome & Scope

Published a CNN–Vision Transformer study in Journal of Voice, reporting approximately 91% accuracy and 0.97 ROC-AUC. Contributed a Bengali voice dataset for research use.

System Architecture

Hybrid CNN-ViT pipeline for voice-based mental health classification

Key Components:

Audio Preprocessing

Noise reduction, normalization, and feature extraction (mel-spectrograms, MFCCs)

CNN Feature Extractor

Convolutional layers capturing local acoustic patterns and spectral features

Vision Transformer

Self-attention mechanism for long-range temporal dependencies in voice sequences

Classification Head

Binary classification of the study labels: stable and unstable.

Attention Visualization

Interpretability module showing which audio segments contribute to predictions

Key Features & Capabilities

Audio preprocessing pipeline with noise reduction and consistent 48 kHz sampling

Log-mel spectrogram conversion (spectrogram-as-image paradigm)

Hybrid CNN-ViT architecture for feature learning

Class imbalance handling with SMOTE

Attention visualization for interpretability

Cross-validation with robust evaluation methodology

Current Scope & Limitations

Research use only. These experiments do not establish a clinically validated diagnostic or screening tool.

Reported scores apply to the study protocol. Split definitions and evaluation details should be checked in the linked paper before comparing results across studies.

Technologies & Tools

PyTorch
Vision Transformer
Librosa
scikit-learn
Weights & Biases
Jupyter