Model Card for "DiagTrast-xlm-roberta-base"
This model is a fine-tuned version of xlm-roberta-base that is a multilingual version of RoBERTa and it is pre-trained on 2.5TB of filtered CommonCrawl data containing 100 languages.
DiagTrast-xlm-roberta-base was trained with hackathon-somos-nlp-2023/DiagTrast dataset to classify statements with each of the 5 selected mental disorders of the DSM-5. While this task is classically approached with neural network-based models, the goal of implementing a transformer model is that instead of basing the classification criteria on keyword search, it is expected to understand natural language through the bidirectional learning of the sentences that the xlm-roberta-base model has.
Uses
The model can be used to classify statements written by professionals who have detected unusual behaviors or characteristics in their patients that would indicate the presence of a mental disorder; at the moment it only provides support for five of the disorders described in the DSM-5. It should be noted that the model aims to identify the predominant disorder, so it would be part of the professional's job to group the symptoms before entering them into the model for cases in which multiple disorders are presumed to be present at the same time.
Direct Use
DiagTrast-xlm-roberta-base is already a fine-tuned model so it could be used directly to classify the statements.
Out-of-Scope Use
This model should not be used as a replacement for a mental health professional because it is always necessary that each situation be evaluated responsibly and using all human intellectual capacity. Initially this model is designed as an auxiliary tool to facilitate the use of the DSM-5 by health professionals.
Bias, Risks, and Limitations
The main limitation of the model is that it is restricted to the identification of only 5 of the DSM-5 disorders.
Recommendations
[More Information Needed]
How to Get Started with the Model
Use the code below to get started with the model.
>>> from transformers import pipeline
>>> classifier = pipeline("text-classification", model='hackathon-somos-nlp-2023/DiagTrast-xlm-roberta-base')
>>> text = ["Gasta más dinero de lo que tiene, a menudo, su falta de control hace que esté en deudas",
"Le gusta estar solo y le molesta la gente a su alrededor, solo piensa en él",
"Tiene pocas habilidades sociales, ignora normas de convivencia",
"Siempre que está en falta, culpa a los demás de sus problemas" ]
>>> classifier.predict(text)
[{'label': 'Trastornos de la personalidad antisocial',
'score': 0.7664140462875366},
{'label': 'Trastornos de la personalidad esquizotípica',
'score': 0.9502732157707214},
{'label': 'Trastornos de la personalidad antisocial',
'score': 0.9722056984901428},
{'label': 'Trastornos de la personalidad antisocial',
'score': 0.49087557196617126}]
Training Details
Training Data
We use the hackathon-somos-nlp-2023/DiagTrast dataset, it was split with 90% of records for the training set and 10% for the test set using the 'datasets' library of hugging face.
Training Procedure
[More Information Needed]
Preprocessing [optional]
[More Information Needed]
Training Hyperparameters
- Training regime: [More Information Needed]
Speeds, Sizes, Times [optional]
[More Information Needed]
Evaluation
Testing Data, Factors & Metrics
Testing Data
[More Information Needed]
Factors
[More Information Needed]
Metrics
[More Information Needed]
Results
[More Information Needed]
Summary
Model Examination [optional]
[More Information Needed]
Environmental Impact
Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).
- Hardware Type: [More Information Needed]
- Hours used: [More Information Needed]
- Cloud Provider: [More Information Needed]
- Compute Region: [More Information Needed]
- Carbon Emitted: [More Information Needed]
Technical Specifications [optional]
Model Architecture and Objective
[More Information Needed]
Compute Infrastructure
[More Information Needed]
Hardware
[More Information Needed]
Software
[More Information Needed]
Citation [optional]
BibTeX:
[More Information Needed]
APA:
[More Information Needed]
Team members
- Downloads last month
- 32