FBANet : Transfer Learning for Depression Recognition Using a Feature-Enhanced Bi-Level Attention Network
The House-Tree-Person (HTP) sketch test is a psychological analysis technique designed to assess the mental health status of test subjects. Nowadays, there are mature methods for the recognition of depression using the HTP sketch test. However, existing works primarily rely on manual analysis of drawing features, which has the drawbacks of strong subjectivity and low automation. Only a small number of works automatically recognize depression using machine learning and deep learning methods, but their complex data preprocessing pipelines and multi-stage computational processes indicate a relatively low level of automation. To overcome the above issues, we present a novel deep learning-based one-stage approach for depression recognition in HTP sketches, which has a simple data preprocessing pipeline and calculation process with a high accuracy rate. In terms of data, we use a hand-drawn HTP sketch dataset, which contains drawings of normal people and patients with depression. In the model aspect, we design a novel network called Feature-Enhanced Bi-Level Attention Network (FBANet), which contains feature enhancement and bi-level attention modules. Due to the limited size of the collected data, transfer learning is employed, where the model is pre-trained on a large-scale sketch dataset and fine-tuned on the HTP sketch dataset. On the HTP sketch dataset, utilizing cross-validation, FBANet achieves a maximum accuracy of 99.07% on the validation dataset, with an average accuracy of 97.71%, outperforming traditional classification models and previous works. In summary, the proposed FBANet, after pre-training, demonstrates superior performance on the HTP sketch dataset and is expected to be a method for the auxiliary diagnosis of depression.
Medienart: |
E-Artikel |
---|
Erscheinungsjahr: |
2023 |
---|---|
Erschienen: |
2023 |
Enthalten in: |
Zur Gesamtaufnahme - volume:25 |
---|---|
Enthalten in: |
Entropy (Basel, Switzerland) - 25(2023), 9 vom: 17. Sept. |
Sprache: |
Englisch |
---|
Beteiligte Personen: |
Wang, Huayi [VerfasserIn] |
---|
Links: |
---|
Themen: |
Bi-level attention |
---|
Anmerkungen: |
Date Revised 03.10.2023 published: Electronic Citation Status Publisher |
---|
doi: |
10.3390/e25091350 |
---|
funding: |
|
---|---|
Förderinstitution / Projekttitel: |
|
PPN (Katalog-ID): |
NLM362579164 |
---|
LEADER | 01000naa a22002652 4500 | ||
---|---|---|---|
001 | NLM362579164 | ||
003 | DE-627 | ||
005 | 20231226091531.0 | ||
007 | cr uuu---uuuuu | ||
008 | 231226s2023 xx |||||o 00| ||eng c | ||
024 | 7 | |a 10.3390/e25091350 |2 doi | |
028 | 5 | 2 | |a pubmed24n1208.xml |
035 | |a (DE-627)NLM362579164 | ||
035 | |a (NLM)37761649 | ||
035 | |a (PII)1350 | ||
040 | |a DE-627 |b ger |c DE-627 |e rakwb | ||
041 | |a eng | ||
100 | 1 | |a Wang, Huayi |e verfasserin |4 aut | |
245 | 1 | 0 | |a FBANet |b Transfer Learning for Depression Recognition Using a Feature-Enhanced Bi-Level Attention Network |
264 | 1 | |c 2023 | |
336 | |a Text |b txt |2 rdacontent | ||
337 | |a ƒaComputermedien |b c |2 rdamedia | ||
338 | |a ƒa Online-Ressource |b cr |2 rdacarrier | ||
500 | |a Date Revised 03.10.2023 | ||
500 | |a published: Electronic | ||
500 | |a Citation Status Publisher | ||
520 | |a The House-Tree-Person (HTP) sketch test is a psychological analysis technique designed to assess the mental health status of test subjects. Nowadays, there are mature methods for the recognition of depression using the HTP sketch test. However, existing works primarily rely on manual analysis of drawing features, which has the drawbacks of strong subjectivity and low automation. Only a small number of works automatically recognize depression using machine learning and deep learning methods, but their complex data preprocessing pipelines and multi-stage computational processes indicate a relatively low level of automation. To overcome the above issues, we present a novel deep learning-based one-stage approach for depression recognition in HTP sketches, which has a simple data preprocessing pipeline and calculation process with a high accuracy rate. In terms of data, we use a hand-drawn HTP sketch dataset, which contains drawings of normal people and patients with depression. In the model aspect, we design a novel network called Feature-Enhanced Bi-Level Attention Network (FBANet), which contains feature enhancement and bi-level attention modules. Due to the limited size of the collected data, transfer learning is employed, where the model is pre-trained on a large-scale sketch dataset and fine-tuned on the HTP sketch dataset. On the HTP sketch dataset, utilizing cross-validation, FBANet achieves a maximum accuracy of 99.07% on the validation dataset, with an average accuracy of 97.71%, outperforming traditional classification models and previous works. In summary, the proposed FBANet, after pre-training, demonstrates superior performance on the HTP sketch dataset and is expected to be a method for the auxiliary diagnosis of depression | ||
650 | 4 | |a Journal Article | |
650 | 4 | |a bi-level attention | |
650 | 4 | |a cross validation | |
650 | 4 | |a depression recognition | |
650 | 4 | |a feature enhancement | |
650 | 4 | |a transfer learning | |
700 | 1 | |a Zhang, Jie |e verfasserin |4 aut | |
700 | 1 | |a Huang, Yaocheng |e verfasserin |4 aut | |
700 | 1 | |a Cai, Bo |e verfasserin |4 aut | |
773 | 0 | 8 | |i Enthalten in |t Entropy (Basel, Switzerland) |d 2008 |g 25(2023), 9 vom: 17. Sept. |w (DE-627)NLM191572098 |x 1099-4300 |7 nnns |
773 | 1 | 8 | |g volume:25 |g year:2023 |g number:9 |g day:17 |g month:09 |
856 | 4 | 0 | |u http://dx.doi.org/10.3390/e25091350 |3 Volltext |
912 | |a GBV_USEFLAG_A | ||
912 | |a GBV_NLM | ||
951 | |a AR | ||
952 | |d 25 |j 2023 |e 9 |b 17 |c 09 |