Downloads · 30 days
9
10% of all-time downloads
ELiRF/Longformer-es-m-base-ICE-MR23-D
Longformer-es-m-base-ICE-MR23-D is a text classification model from ELiRF. Use it when you need a label for a piece of text. It is set up for transformers.
Longformer-es-m-base-ICE-MR23-D is a Spanish long-context language model for early detection of depression risk, trained using the Incremental Context Expansion (ICE) methodology. The model was developed for the Menta…
Downloads · 30 days
9
10% of all-time downloads
All-time downloads
89
Public
Parameters
149M
5.9 GB on disk
Likes
1
Public
Click a slice to open those files.
.safetensors595 MB · 99%
From the Hugging Face model README
Longformer-es-m-base-ICE-MR23-D is a Spanish long-context language model for early detection of depression risk, trained using the Incremental Context Expansion (ICE) methodology. The model was developed for the MentalRisk 2023 Depression (MR23-D) shared task and builds upon the Longformer-es-mental-base foundation model.
This model corresponds to the base-sized version of the ICE-adapted Longformer models. Compared to its large counterpart, it contains fewer parameters, offering a more computationally efficient alternative while maintaining the ability to process long user message histories.
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. This training strategy better reflects real-world early detection scenarios, where predictions must be generated with limited and evolving evidence.
The model is based on the Longformer architecture and supports input sequences of up to 4096 tokens, allowing it to integrate evidence distributed across long and heterogeneous user histories. It has been fine-tuned to detect depression-related risk signals from Spanish user-generated content under both early detection and full-context evaluation settings.
This model is intended for research purposes in early mental health risk detection, specifically for depression.
The model can be used directly for early detection of depression risk from Spanish user-generated content, generating predictions incrementally as new user messages become available.
Incremental Context Expansion (ICE) is a training methodology designed for early and incremental mental health detection tasks. Instead of training models on full user histories, ICE generates multiple incremental contexts per user, each corresponding to a partial message history.
This approach allows the model to:
ICE modifies the dataset construction process while keeping the standard fine-tuning pipeline unchanged.
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("ELiRF/Longformer-es-m-base-ICE-MR23-D")
model = AutoModelForSequenceClassification.from_pretrained(
"ELiRF/Longformer-es-m-base-ICE-MR23-D"
)
inputs = tokenizer(
"Ejemplo de historial de mensajes relacionado con depresión.",
return_tensors="pt",
truncation=True,
max_length=4096
)
outputs = model(**inputs)
The model was fine-tuned on the MentalRisk 2023 Depression (MR23-D) dataset. Training data was restructured using the ICE methodology, generating incremental user contexts from original user timelines.
When evaluated on the MentalRisk 2023 Depression task, Longformer-es-m-base-ICE-MR23-D shows competitive performance under both early detection and full-context (user-level) evaluation settings, offering a favorable trade-off between performance and computational efficiency.
This model is part of an ongoing research project.
The associated paper is currently under review and will be added to this model card once the publication process is completed.
ELiRF research group (VRAIN, Universitat Politècnica de València)