driwnet commited on
Commit
4bd8a90
·
verified ·
1 Parent(s): 0c2ec8c

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +141 -0
README.md ADDED
@@ -0,0 +1,141 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ tags:
4
+ - spanish
5
+ - mental-health
6
+ - longformer
7
+ - early-detection
8
+ - eating-disorder
9
+ - ice
10
+ - nlp
11
+ language:
12
+ - es
13
+ base_model:
14
+ - ELiRF/Longformer-es-mental-base
15
+ ---
16
+ # Model Card for Longformer-es-m-base-ICE-MR23-ED
17
+
18
+ ## Model Description
19
+
20
+ Longformer-es-m-base-ICE-MR23-ED is a Spanish long-context language model for **early detection of eating disorder risk**, trained using the **Incremental Context Expansion (ICE)** methodology.
21
+ This model builds upon the **Longformer-es-mental-base** foundation model and represents the **base-sized version** of the ICE-adapted Longformer models for the Eating Disorder task.
22
+
23
+ Compared to its large counterpart, this version contains **fewer parameters**, offering a more computationally efficient alternative while preserving the ability to process long user message histories.
24
+ The model is designed for scenarios where mental health–related evidence is distributed across multiple messages and predictions must be generated incrementally.
25
+
26
+ The ICE methodology restructures the training data at the **context level**, enabling the model to learn from progressively expanding user message histories rather than complete user timelines.
27
+ This setup better reflects real-world early detection conditions.
28
+
29
+ The model is based on the Longformer architecture and supports input sequences of up to **4096 tokens**.
30
+ It has been fine-tuned for the **Eating Disorder (ED)** task using the **MentalRisk 2023 (MR23)** benchmark under early detection evaluation settings.
31
+
32
+ - Developed by: ELiRF group, VRAIN (Valencian Research Institute for Artificial Intelligence), Universitat Politècnica de València
33
+ - Shared by: ELiRF
34
+ - Model type: Transformer-based sequence classification model (Longformer)
35
+ - Language: Spanish
36
+ - Base model: Longformer-es-mental-base
37
+ - License: Same as base model
38
+
39
+ ## Uses
40
+
41
+ This model is intended for **research purposes** in early mental health risk detection.
42
+
43
+ ### Direct Use
44
+
45
+ The model can be used directly for **early detection of eating disorder risk** from Spanish user-generated content, where predictions are generated incrementally as new user messages become available.
46
+
47
+ ### Downstream Use
48
+
49
+ - Early risk detection for eating disorders
50
+ - User-level mental health screening
51
+ - Comparative studies of early detection methodologies
52
+ - Research on incremental and temporally-aware NLP models
53
+
54
+ ### Out-of-Scope Use
55
+
56
+ - Automated intervention systems without human supervision
57
+ - Use on languages other than Spanish
58
+ - High-stakes or real-time decision-making affecting individuals’ health
59
+
60
+ ## ICE Methodology
61
+
62
+ Incremental Context Expansion (ICE) is a training methodology designed for early detection tasks.
63
+ Instead of training on full user histories, ICE generates **multiple incremental contexts per user**, each corresponding to a partial message history.
64
+
65
+ This approach allows the model to:
66
+ - Learn from early and incomplete evidence
67
+ - Reduce detection latency
68
+ - Improve robustness under early detection evaluation metrics
69
+
70
+ ICE modifies the dataset construction process while keeping the standard fine-tuning pipeline unchanged.
71
+
72
+ ## Bias, Risks, and Limitations
73
+
74
+ - Training data originates from social media platforms and may contain demographic and cultural biases.
75
+ - Automatically translated texts may include translation artifacts.
76
+ - Early detection tasks are inherently uncertain due to limited available evidence.
77
+ - The model does not provide explanations or clinical interpretations of its predictions.
78
+
79
+ ## How to Get Started with the Model
80
+
81
+ ```python
82
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
83
+
84
+ tokenizer = AutoTokenizer.from_pretrained("ELiRF/Longformer-es-m-base-ICE-MR23-ED")
85
+ model = AutoModelForSequenceClassification.from_pretrained(
86
+ "ELiRF/Longformer-es-m-base-ICE-MR23-ED"
87
+ )
88
+
89
+ inputs = tokenizer(
90
+ "Ejemplo de historial de mensajes relacionado con trastornos alimentarios.",
91
+ return_tensors="pt",
92
+ truncation=True,
93
+ max_length=4096
94
+ )
95
+
96
+ outputs = model(**inputs)
97
+ ```
98
+
99
+ ## Training Details
100
+
101
+ ### Training Data
102
+
103
+ The model was fine-tuned on the **MentalRisk 2023 Eating Disorder (MR23-ED)** dataset.
104
+ Training data was restructured using the **ICE methodology**, generating incremental user contexts from original user timelines.
105
+
106
+ ### Training Procedure
107
+
108
+ - Base model: Longformer-es-mental-base
109
+ - Fine-tuning strategy: ICE-based context-level training
110
+ - Objective: Sequence classification
111
+ - Training regime: fp16 mixed precision
112
+
113
+ ## Evaluation
114
+
115
+ ### Results
116
+
117
+ When evaluated on the MentalRisk 2023 Eating Disorder task, Longformer-es-m-base-ICE-MR23-ED shows **competitive performance** under **early detection** evaluation settings and strong results in **full-context (user-level)** scenarios, offering a favorable trade-off between performance and computational efficiency.
118
+
119
+ ## Environmental Impact
120
+
121
+ - Hardware type: NVIDIA A40 GPUs
122
+ - Training time: several hours (fine-tuning)
123
+
124
+ ## Technical Specifications
125
+
126
+ ### Model Architecture and Objective
127
+
128
+ - Architecture: Longformer (base)
129
+ - Objective: Sequence classification
130
+ - Maximum sequence length: 4096 tokens
131
+ - Model size: approximately 150M parameters
132
+
133
+ ## Citation
134
+
135
+ This model is part of an ongoing research project.
136
+ The associated paper is currently under review and will be added to this model card once the publication process is completed.
137
+
138
+
139
+ ## Model Card Authors
140
+
141
+ ELiRF research group (VRAIN, Universitat Politècnica de València)