Update README.md
Browse files
README.md
CHANGED
|
@@ -24,21 +24,18 @@ tags:
|
|
| 24 |
|
| 25 |
This model was developed to support standardized, scalable mental health assessments in both clinical and low-resource settings.
|
| 26 |
|
| 27 |
-
---
|
| 28 |
-
|
| 29 |
|
| 30 |
-
##
|
| 31 |
|
| 32 |
- **Base model**: `bert-base-german-cased`
|
| 33 |
-
- **Task**: Ordinal regression
|
| 34 |
-
- **Language**: German
|
| 35 |
- **Input**: Text (dialogue segment grouped by MADRS topic)
|
| 36 |
- **Output**: Predicted score for each MADRS item (rounded integer 0–6)
|
| 37 |
- **Training data**: Mix of real and synthetic clinician–patient interviews (MADRS-structured)
|
| 38 |
|
| 39 |
-
---
|
| 40 |
|
| 41 |
-
##
|
| 42 |
|
| 43 |
This model is intended for research and development use. It is not a certified medical device. The goal is to:
|
| 44 |
- Explore AI-assisted symptom severity assessment
|
|
@@ -88,7 +85,7 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
|
| 88 |
model.eval().to("cuda" if torch.cuda.is_available() else "cpu")
|
| 89 |
```
|
| 90 |
|
| 91 |
-
###
|
| 92 |
Assume you have a conversation log like this:
|
| 93 |
|
| 94 |
```python
|
|
@@ -113,20 +110,17 @@ print(scores)
|
|
| 113 |
|
| 114 |
---
|
| 115 |
|
| 116 |
-
##
|
| 117 |
|
| 118 |
Model trained and released by [Samantha Weber](https://github.com/webersamantha). Research conducted as part of efforts to improve AI-driven mental health tools. Thanks to all clinicians and collaborators who contributed to the annotated MADRS dataset.
|
| 119 |
|
| 120 |
-
---
|
| 121 |
-
|
| 122 |
|
| 123 |
-
##
|
| 124 |
|
| 125 |
The model was evaluated on a held-out clinical validation set and achieved strong performance under both strict and flexible scoring criteria (±1 deviation tolerance). See publication for full metrics.
|
| 126 |
|
| 127 |
-
---
|
| 128 |
|
| 129 |
-
##
|
| 130 |
|
| 131 |
If you use this model, please cite:
|
| 132 |
> Weber, S. et al. (2025). "Using a Fine-tuned Large Language Model for Symptom-based Depression Evaluation" *Preprint*. https://doi.org/10.21203/rs.3.rs-6555767/v1
|
|
|
|
| 24 |
|
| 25 |
This model was developed to support standardized, scalable mental health assessments in both clinical and low-resource settings.
|
| 26 |
|
|
|
|
|
|
|
| 27 |
|
| 28 |
+
## Model Details
|
| 29 |
|
| 30 |
- **Base model**: `bert-base-german-cased`
|
| 31 |
+
- **Task**: Ordinal regression (scores 0–6)
|
| 32 |
+
- **Language**: German
|
| 33 |
- **Input**: Text (dialogue segment grouped by MADRS topic)
|
| 34 |
- **Output**: Predicted score for each MADRS item (rounded integer 0–6)
|
| 35 |
- **Training data**: Mix of real and synthetic clinician–patient interviews (MADRS-structured)
|
| 36 |
|
|
|
|
| 37 |
|
| 38 |
+
## Intended Use
|
| 39 |
|
| 40 |
This model is intended for research and development use. It is not a certified medical device. The goal is to:
|
| 41 |
- Explore AI-assisted symptom severity assessment
|
|
|
|
| 85 |
model.eval().to("cuda" if torch.cuda.is_available() else "cpu")
|
| 86 |
```
|
| 87 |
|
| 88 |
+
### Predict on a full structured interview / Run inference:
|
| 89 |
Assume you have a conversation log like this:
|
| 90 |
|
| 91 |
```python
|
|
|
|
| 110 |
|
| 111 |
---
|
| 112 |
|
| 113 |
+
## Acknowledgements
|
| 114 |
|
| 115 |
Model trained and released by [Samantha Weber](https://github.com/webersamantha). Research conducted as part of efforts to improve AI-driven mental health tools. Thanks to all clinicians and collaborators who contributed to the annotated MADRS dataset.
|
| 116 |
|
|
|
|
|
|
|
| 117 |
|
| 118 |
+
## Evaluation
|
| 119 |
|
| 120 |
The model was evaluated on a held-out clinical validation set and achieved strong performance under both strict and flexible scoring criteria (±1 deviation tolerance). See publication for full metrics.
|
| 121 |
|
|
|
|
| 122 |
|
| 123 |
+
## Citation
|
| 124 |
|
| 125 |
If you use this model, please cite:
|
| 126 |
> Weber, S. et al. (2025). "Using a Fine-tuned Large Language Model for Symptom-based Depression Evaluation" *Preprint*. https://doi.org/10.21203/rs.3.rs-6555767/v1
|