$$\rightleftharpoonup{xx}$$
$$\longleftharp{xx}$$,
$$\longrightharp{xx}$$,
Vorgeschlagene Methodik
Diese Studie schlägt einen generativen, KI-basierten Rahmen zur Erkennung von COVID-19 anhand von Hustensignalen vor. Das Framework integriert generative Modelle mit diskriminierenden Klassifikatoren, wobei Trainings- und Evaluationsdaten strikt getrennt bleiben. Abbildung 1 zeigt die detaillierte Architektur des vorgeschlagenen GAN–VAE–ADCNN-Frameworks und zeigt den Fluss von Audio-Vorverarbeitung und Feature-Extraktion über latentes Repräsentationslernen über einen Variational Autoencoder (VAE), synthetische Feature-Generierung mit einem Generative Adversarial Network (GAN) und die endgültige Klassifikation mit Deep Convolutional Neural Networks (DCNN) und Attention-based DCNN (ADCNN). Die Abbildung zeigt, dass generative Komponenten nur während des Trainings verwendet werden, während die Klassifikation mit diskriminativen Modellen durchgeführt wird.

Abbildung 1: Überblick über die GAN–VAE–ADCNN-Architektur. Schematisch der vorgeschlagenen hybriden Pipeline, bei der hustenbasierte akustische Merkmale mit einem VAE codiert, durch GAN-basierte synthetische Sample-Generierung ergänzt und mit DCNN- und aufmerksamkeitsbasierten DCNN-(ADCNN)-Modellen klassifiziert werden. Bitte klicken Sie hier, um eine größere Version dieser Abbildung anzusehen.
Modellarchitektur und Implementierung
Die in diesem Rahmen verwendeten generativen und diskriminierenden Modelle wurden mit festen und reproduzierbaren Architekturen implementiert. Das GAN besteht aus einem Generator mit drei vollständig verbundenen Schichten (128, 256 und 512 Einheiten) mit ReLU-Aktivierung und einem Diskriminator mit drei vollständig verbundenen Schichten (512, 256 und 128 Einheiten) mit LeakyReLU-Aktivierung, gefolgt von einem Sigma-Ausgang.
Der VAE-Encoder besteht aus zwei vollständig verbundenen Schichten mit 256 und 128 Einheiten, gefolgt von der Schätzung des latenten Mittelwerts und der Varianz mit einer latenten Dimension von 64. Der Decoder spiegelt diese Struktur wider und wird mit einer Kombination aus Rekonstruktionsverlust und Kullback–Leibler-Divergenz trainiert, um einen strukturierten und probabilistischen latenten Raum zu lernen.
Der DCNN-Klassifikator umfasst vier Faltungsblöcke mit 3x3-Kernen sowie 32, 64, 128 bzw. 256 Filtern. Auf jeden Block folgt eine Batch-Normalisierung, eine ReLU-Aktivierung und ein 2x2-Max-Pooling. Das ADCNN erweitert das DCNN, indem es nach dem finalen Faltungsblock einen kanalweisen Aufmerksamkeitsmechanismus einführt. Alle Modelle wurden mit dem Adam-Optimierer optimiert, mit einer Lernrate von 0,0001 und einer Batchgröße von 32. Wie in Abbildung 1 gezeigt, operieren VAE und GAN nur während der Ausbildung, während DCNN und ADCNN für die Klassifizierung verwendet werden. Das vollständige Trainings- und Bewertungsverfahren ist in Algorithmus 1 zusammengefasst.
Algorithmus 1: GAN–VAE-basierter COVID-19-Nachweisrahmen
Eingabe: Vorverarbeitete Husten-Audioaufnahmen
Ausgabe: Binärklassifizierungsetikett (COVID-19 positiv/negativ)
Die Ausbildungs- und Evaluationsverfahren folgten einer festen, reproduzierbaren Pipeline. Alle Husten-Audioaufnahmen wurden auf 16 kHz neu abgetastet, Rauschunterdrückung unterzogen und die Amplitude normalisiert. Die Aufnahmen wurden in Segmente mit fester Länge unterteilt, aus denen MFCC, Chroma und spektrale Merkmale extrahiert wurden. Aus jedem Audiosegment wurden insgesamt 40 Mel-Frequenz-Cepstralkoeffizienten (MFCCs) extrahiert. Zusätzlich wurden 12 Chroma-Merkmale berechnet. Die resultierende Darstellung bildet für jedes Hustensegment einen 52-dimensionalen Merkmalsvektor. Es wurde eine Subjektebene-Aufteilung durchgeführt, um die Daten in Trainings-, Validierungs- und Testsets zu unterteilen. Der VAE wurde darauf trainiert, probabilistische latente Darstellungen zu lernen, und die daraus resultierenden latenten Vektoren wurden verwendet, um das GAN für die Generierung synthetischer Merkmalsmerkmale zu trainieren. Der Trainingsdatensatz wurde mithilfe von GAN–VAE erzeugter Stichproben erweitert, während synthetische Daten von der Validierung und Prüfung ausgeschlossen wurden. DCNN- und ADCNN-Klassifikatoren wurden mit den erweiterten Trainingsdaten trainiert, und die Endbewertung erfolgte an der ausgewiesenen Testreihe unter Verwendung standardisierter Leistungskennzahlen.
Trainingsaufbau, Datenaufteilung und Leckvermeidung
Alle Experimente wurden mit einer festen, reproduzierbaren Trainingskonfiguration durchgeführt. Die Datenaufteilung wurde auf Subjektebene vor der Audiosegmentierung durchgeführt, um Datenlecks zu verhindern. Der Datensatz wurde in Trainings-, Validierungs- und Testsets im Verhältnis 80:10:10 unterteilt, sodass alle Segmente derselben Aufzeichnung einer einzigen Teilmenge zugeordnet wurden. Der Trainingssatz wurde für Modelllernen und generative Datenerweiterung verwendet, der Validierungssatz für Hyperparameter-Tuning und frühes Stoppen, und der Testsatz wurde für die endgültige Leistungsbewertung bereitgestellt. Synthetische Proben, die vom GAN–VAE-Rahmen generiert wurden, wurden ausschließlich während der Ausbildung verwendet und streng von der Validierung und Prüfung ausgeschlossen, um eine faire Bewertung zu gewährleisten.
Die Modelle wurden für maximal 100 Epochen trainiert, mit frühzeitigem Stoppen aufgrund von Validierungsverlust und einer Geduld von 10 Epochen. Die Optimierung erfolgte mit dem Adam-Optimierer mit einer Lernrate von 0,0001 und einer Batchgröße von 32. Binäre Kreuzentropieverluste wurden für die Klassifikation angewendet, während Rekonstruktion und Adversarialverluste zum Training der VAE- bzw. GAN-Komponenten verwendet wurden. Dieses einheitliche Trainings- und Bewertungsprotokoll gewährleistet die Reproduzierbarkeit, verhindert Datenverluste und sorgt für eine strikte Trennung zwischen Trainings- und Evaluierungsphasen.
Datensatzbeschreibung
Zwei öffentlich zugängliche Husten-Audio-Datensätze – COUGHVID und Virufy – wurden verwendet, um eine großflächige akustische Vielfalt mit klinisch referenzierten Labels auszubalancieren, mit klar getrennten Rollen in Training und Evaluation.
COUGHVID ist eine durch die Crowdsourcing erstellte Sammlung von Hustenaufnahmen mit teilweisen Expertenannotationen und selbstberichteten Metadaten. Um die Datenqualität zu gewährleisten, wurden 428 Aufnahmen nach Anwendung vordefinierter Qualitätskontrollkriterien ausgewählt, darunter minimale Signaldauer, angemessenes Signal-Rausch-Verhältnis, Entfernung beschädigter oder mehrdeutiger Proben sowie das Vorhandensein identifizierbarer Hustenanfälle mit Symptommetadaten. Nach der Vorverarbeitung und Segmentierung ergaben diese Aufnahmen 1.719 Hustensegmente, die auf das WAV-Format mit einer Abtastrate von 16 kHz standardisiert wurden. COUGHVID wurde verwendet, um sowohl die generativen Modelle als auch die diskriminativen Klassifikatoren zu trainieren.
Virufy besteht aus von Ärzten überwachten Hustenaufnahmen, die als RT-PCR positiv oder negativ auf COVID-19 gekennzeichnet sind. Alle 16 verfügbaren Aufnahmen waren enthalten, was nach der Segmentierung zu 234 Hustensegmenten führte, ebenfalls standardisiert auf 16 kHz. Virufy wurde ausschließlich für externe Cross-Dataset-Bewertung zur Bewertung der Generalisierungsleistung verwendet und nicht für Training, Feinabstimmung oder generative Augmentation.
Der vorgeschlagene Rahmen sollte daher als Proof-of-Concept-Screening-Ansatz interpretiert werden, der unter kontrollierten experimentellen Bedingungen bewertet wird, und nicht als klinisch validiertes Diagnosesystem.
Datenvorverarbeitung und Segmentierung
Alle Aufnahmen wurden auf 16 kHz neu abgetastet, in Mono umgewandelt und der Stillstandsentfernung, spektraler Rauschreduktion und Amplitudennormalisierung unterzogen. Die Segmentierung erfolgte nach der Aufteilung auf Subjektebene, um Datenlecks zu verhindern. Jede Aufnahme wurde in überlappende 2–4 Segmente unterteilt, und niedrigenergetischen oder stummen Segmente wurden verworfen.
Externes Validierungsprotokoll
Die externe Validierung erfolgte durch Datensatzüberprüfung. Modelle, die mit COUGHVID trainiert wurden, wurden mit Virufy zur externen Validierung evaluiert. Dieses Protokoll bewertet die Verallgemeinerung über Datensätze, die unter verschiedenen Bedingungen gesammelt wurden, anstatt auf prospektive klinische Validierung. Abbildung 2 bietet einen protokollübergreifenden Überblick über diesen Workflow und veranschaulicht die Nutzung von Datensätzen, die Vorverarbeitung, die Feature-Extraktion, das Training von Klassifikatoren und die Evaluationsszenarien. Während Abbildung 1 sich auf die interne Modellarchitektur konzentriert, legt Abbildung 2 den Schwerpunkt auf experimentelles Design und Datenfluss über Trainings- und Testphasen hinweg.

Abbildung 2: Arbeitsablauf des vorgeschlagenen COVID-19-Hustenscreening-Frameworks. Illustration der vollständigen Verarbeitungspipeline, einschließlich Vorverarbeitung, Merkmalsextraktion (MFCC, Chroma, spektrale Merkmale), GAN–VAE-basiertes Darstellungslernen und -augmentation (nur Training) sowie die endgültige Klassifikation unter subjektebenerder Aufteilung und Datensatz-übergreifender Bewertung. Bitte klicken Sie hier, um eine größere Version dieser Abbildung anzusehen.