VIS-DICT: A Visual Dictionary for Missing Modality Imputation in Social Network Depression Detection
Authors: Hamed Marvi, Mohammad Mehdi Keikha, Abolfazl Nadi
Organizations: Department of Computer Science, School of Mathematics, Statistics and Computer Science, College of Science, University of Tehran, Tehran, Iran · Department of Computer Science, University of Sistan and Baluchestan, Zahedan, Iran
Tracking social media posts can help spot early signs of depression. Recent studies show that combining text and images works better for detecting depression than using text alone. However, many social media posts do not have images, which makes it hard to use multimodal models. Most existing methods fill in missing images using retrieval or generative models that need extra training. In this paper, we introduce Vis-Dict, a dictionary-based method that builds missing visual features by linking words to average image vectors from complete training posts. These estimated visual features are then combined with text to track changes in user behavior over time. We tested Vis-Dict on a social media dataset using user timelines of up to 512 posts and compared it with other missing-data methods. The results show that Vis-Dict performs on par with generative networks, reaching an F1-score of 0.9454 and an ROC-AUC of 0.9890. Most importantly, Vis-Dict achieves this strong performance with zero trainable parameters for image generation. These findings show that directly connecting words to visual features is an effective and practical way to handle missing images in depression detection systems.
Figures & tables
Figure 1. Overall architecture of the tweet-level pipeline. The system extracts text and image representations ( htxt,himg∈R768 ), estimates missing visual embeddings using five alternative imputation methods when an image is absent, projects both representations to R512 , and concatenates them into a unified vector ( zfused∈R1024 ) for single-tweet classification and user-level sequence modeling.
Figure 2. Architecture of the textual feature extraction pipeline using XLM-RoBERTa [ 32 ] to generate a 768-dimensional dense vector.
Figure 3. Architecture of the visual feature extraction pipeline using CLIP ViT [ 42 ] , illustrating the custom resizing and average-color padding strategy.
Figure 4. Architecture of the Text-to-Visual (T2V) translation module.
Figure 5. Overview of the Vis-Dict framework: dictionary construction from complete multimodal posts (Step 1) and feature imputation for posts without images (Step 2).
Split
Class 0 (Normal)
Class 1 (Depressed)
Avg Tweets / User
Missing Images
Training
1,401
1,401
235
90.6%
Validation
175
175
202
88.7%
Testing
176
176
290
90.1%
Overall
1,752
1,752
237
90.4%
Table 1. User-level dataset distribution and longitudinal timeline statistics across the experimental splits.
Figure 7. Softmax Averaging: Depressed F1 -Score across various classification thresholds. Performance peaks at θ=0.5 across all imputation strategies.
Imputation Strategy
Training Paradigm
Depressed F1
ROC-AUC
Zero Placeholder
Unjoint
0.9111
0.9091
Joint
0.9061
0.9034
Constant Learnt
Unjoint
0.9136
0.9119
Joint
0.9036
0.9006
MissRAG
Unjoint
0.9183
0.9176
Joint
0.9076
0.9062
Table 8. User-level classification results using Softmax Probability Averaging (Threshold θ=0.5 ).
Model Configuration
Training Paradigm
Depressed F1
ROC-AUC
1. Unimodal Baselines
Text-Only (All Samples)
Frozen
0.9359±0.0127
0.9869±0.0005
Text-Only (Full Modality)
Frozen
0.8915±0.0023
0.9683±0.0007
Image-Only (Full Modality)
Unjoint
0.8611±0.0234
0.9361±0.0048
Image-Only (Full Modality
Joint
0.7747±0.0506
0.8707±0.0041
2. Multimodal (w/ Imputation)
Table 9. User-level classification performance using the Sequential Transformer across 5 runs.
Imputation Strategy
Positional Encoding
Depressed F1
ROC-AUC
Vis-Dict (Unjoint)
Included
0.9454±0.0128
0.9890±0.0009
Vis-Dict (Unjoint)
Removed
0.9416±0.0102
0.9897±0.0010
T2V Network (Joint)
Included
0.9468±0.0048
0.9855±0.0008
T2V Network (Joint)
Removed
0.9024±0.0229
0.9762±0.0057
MissRAG (Joint)
Included
0.9408±0.0075
0.9855±0.0008
MissRAG (Joint)
Removed
0.9249±0.0080
0.9825±0.0011
Table 10. Sequential Transformer performance with and without Positional Encoding (5 runs).
Imputation Strategy
Training Paradigm
Depressed F1
ROC-AUC
1. Full Modality (Real Images)
Real Image (Oracle)
Unjoint
0.9062±0.0014
0.9698±0.0022
Real Image (Oracle)
Joint
0.8684±0.0104
0.9475±0.0050
2. Missing Modality (Imputed)
MissRAG
Unjoint
0.9423±0.0062
0.9841±0.0002
MissRAG
Joint
0.9425±0.0057
0.9836±0.0007
Table 11. Sequential model performance separated by Full Modality and Missing Modality subsets (5 runs).
College of Information Science and Engineering, Ritsumeikan University, Shiga, Japan · College of Computer Science and Technology, Zhejiang University, Hangzhou, China · Faculty of Engineering, University of the Ryukyus, Japan