$$\rightleftharpoonup{xx}$$
$$\longleftharp{xx}$$,
$$\longrightharp{xx}$$,
Voorgestelde methodologie
Deze studie stelt een generatief AI-gebaseerd kader voor om COVID-19 te detecteren aan de hand van hoestsignalen. Het framework integreert generatieve modellen met discriminerende classifiers, waarbij trainings- en evaluatiegegevens strikt gescheiden worden gehouden. Figuur 1 illustreert de gedetailleerde architectuur van het voorgestelde GAN–VAE–ADCNN-framework, en toont de stroom van audio-preprocessing en feature-extractie via latent representation learning via een Variational Autoencoder (VAE), synthetische feature-generatie met een Generative Adversarial Network (GAN), en de uiteindelijke classificatie met Deep Convolutional Neural Networks (DCNN) en Attention-based DCNN (ADCNN). De figuur toont aan dat generatieve componenten alleen tijdens training worden gebruikt, terwijl classificatie wordt uitgevoerd met behulp van discriminatiemodellen.

Figuur 1: GAN–VAE–ADCNN architectuuroverzicht. Schema van de voorgestelde hybride pijplijn, waarbij hoest-afgeleide akoestische kenmerken worden gecodeerd met een VAE, aangevuld met GAN-gebaseerde synthetische monstergeneratie en geclassificeerd met DCNN- en aandachtgebaseerde DCNN (ADCNN)-modellen. Klik hier om een grotere versie van deze figuur te bekijken.
Modelarchitectuur en implementatie
De generatieve en discriminatieve modellen die in dit kader worden gebruikt, zijn geïmplementeerd met vaste en reproduceerbare architecturen. De GAN bestaat uit een generator met drie volledig verbonden lagen (128, 256 en 512 eenheden) die ReLU-activatie gebruiken, en een discriminator met drie volledig verbonden lagen (512, 256 en 128 eenheden) met LeakyReLU-activatie gevolgd door een sigmoid-uitvoer.
De VAE-encoder bestaat uit twee volledig verbonden lagen met 256 en 128 eenheden, gevolgd door schatting van het latente gemiddelde en de variantie met een latente dimensionaliteit van 64. De decoder weerspiegelt deze structuur en wordt getraind met een combinatie van reconstructieverlies en Kullback–Leibler-divergentie om een gestructureerde en probabilistische latente ruimte te leren.
De DCNN-classifier bevat vier convolutionele blokken met respectievelijk 3x3 kernen en 32, 64, 128 en 256 filters. Elk blok wordt gevolgd door batchnormalisatie, een ReLU-activatie en 2x2 max pooling. De ADCNN breidt de DCNN uit door een kanaalgewijze aandachtsmechanisme te integreren na het laatste convolutionele blok. Alle modellen werden geoptimaliseerd met de Adam-optimizer met een leersnelheid van 0,0001 en een batchgrootte van 32. Zoals weergegeven in Figuur 1, opereren de VAE en GAN alleen tijdens de training, terwijl DCNN en ADCNN worden gebruikt voor classificatie. De volledige trainings- en evaluatieprocedure wordt samengevat in Algoritme 1.
Algoritme 1: GAN–VAE-gebaseerd COVID-19 detectiekader
Input: Voorbewerkte hoestgeluidsopnamen
Output: Binaire classificatielabel (COVID-19 positief/negatief)
De trainings- en evaluatieprocedures volgden een vaste, reproduceerbare pijplijn. Alle hoestgeluidsopnamen werden opnieuw gesampled tot 16 kHz, onderworpen aan ruisreductie en amplitude genormaliseerd. De opnames werden gesegmenteerd in segmenten van vaste lengte, waaruit MFCC, chroma en spectrale kenmerken werden geëxtraheerd. In totaal werden 40 Mel-Frequency Cepstral Coëfficiënten (MFCC's) uit elk audiosegment geëxtraheerd. Daarnaast werden 12 chroma-kenmerken berekend. De resulterende representatie vormt een 52-dimensionale kenmerkvector voor elk hoestsegment. Er werd een splitsing op proefpersoonsniveau uitgevoerd om de gegevens te verdelen in trainings-, validatie- en testsets. De VAE werd getraind om probabilistische latente representaties te leren, en de resulterende latente vectoren werden gebruikt om de GAN te trainen voor synthetische featuregeneratie. De trainingsdataset werd uitgebreid met GAN–VAE-gegenereerde steekproeven, terwijl synthetische data werden uitgesloten van validatie en testen. DCNN- en ADCNN-classifiers werden getraind op de augmented trainingsgegevens, en de eindevaluatie werd uitgevoerd op de vastgehouden testset met behulp van standaard prestatie-metrics.
Trainingsopzet, datasplitsing en lekpreventie
Alle experimenten werden uitgevoerd met een vaste, reproduceerbare trainingsconfiguratie. Datasplitsing werd uitgevoerd op proefpersoonniveau vóór audiosegmentatie om datalekkage te voorkomen. De dataset werd verdeeld in trainings-, validatie- en testsets in een verhouding van 80:10:10, zodat alle segmenten uit dezelfde opname aan één subset werden toegewezen. De trainingsset werd gebruikt voor modelleren en generatieve data-augmentatie, de validatieset voor hyperparameterafstemming en vroegtijdig stoppen, en de testset werd vastgelegd voor de definitieve prestatie-evaluatie. Synthetische monsters die door het GAN–VAE-kader werden gegenereerd, werden exclusief gebruikt tijdens de training en werden strikt uitgesloten van validatie en testen om een eerlijke evaluatie te waarborgen.
Modellen werden getraind voor maximaal 100 epochs, met vroegtijdige stopzetten op basis van validatieverlies en een geduld van 10 epochs. Optimalisatie werd uitgevoerd met de Adam-optimizer met een leersnelheid van 0,0001 en een batchgrootte van 32. Binair kruis-entropieverlies werd toegepast voor classificatie, terwijl reconstructie en adversariale verliezen werden gebruikt voor het trainen van respectievelijk de VAE- en GAN-componenten. Dit uniforme trainings- en evaluatieprotocol zorgt voor reproduceerbaarheid, voorkomt datalekken en handhaaft een strikte scheiding tussen trainings- en evaluatiefasen.
Datasetbeschrijving
Twee publiek beschikbare hoestaudiodatasets—COUGHVID en Virufy—werden gebruikt om grootschalige akoestische diversiteit in balans te brengen met klinisch geraadpleegde labels, met duidelijk gescheiden rollen in training en evaluatie.
COUGHVID is een crowdsourced verzameling hoestopnames met gedeeltelijke expertannotatie en zelfgerapporteerde metadata. Om de datakwaliteit te waarborgen, werden 428 opnames geselecteerd na toepassing van vooraf gedefinieerde kwaliteitscontrolecriteria, waaronder minimale signaalduur, een adequate signaal-ruisverhouding, verwijdering van corrupte of ambigue monsters en de aanwezigheid van identificeerbare hoestincidenten met symptoommetadata. Na preprocessing en segmentatie leverden deze opnames 1.719 hoestsegmenten op, gestandaardiseerd naar WAV-formaat met een samplefrequentie van 16 kHz. COUGHVID werd gebruikt voor het trainen van zowel generatieve modellen als discriminatieve classifiers.
Virufy bestaat uit door artsen begeleide hoestopnames die als RT-PCR positief of negatief zijn gelabeld voor COVID-19. Alle 16 beschikbare opnames werden opgenomen, wat resulteerde in 234 hoestsegmenten na segmentatie, eveneens gestandaardiseerd op 16 kHz. Virufy werd uitsluitend gebruikt voor externe cross-dataset-evaluatie om de generalisatieprestaties te beoordelen en werd niet gebruikt voor training, fine-tuning of generatieve augmentatie.
Het voorgestelde kader moet daarom worden geïnterpreteerd als een proof-of-concept screeningsbenadering geëvalueerd onder gecontroleerde experimentele omstandigheden, in plaats van als een klinisch gevalideerd diagnostisch systeem.
Voorverwerking en segmentatie van gegevens
Alle opnames werden opnieuw gesampled naar 16 kHz, omgezet naar mono en onderworpen aan stilteverwijdering, spectrale ruisreductie en amplitudenormalisatie. Segmentatie werd uitgevoerd na splitsing op proefpersoonsniveau om datalekkage te voorkomen. Elke opname werd verdeeld in overlappende segmenten van 2–4 s, en laag-energetische of stille segmenten werden weggegooid.
Extern Validatieprotocol
Externe validatie werd uitgevoerd door middel van cross-dataset-evaluatie. Modellen die getraind zijn met COUGHVID werden geëvalueerd op Virufy voor externe validatie. Dit protocol beoordeelt generalisatie over datasets die onder verschillende omstandigheden zijn verzameld, in plaats van prospectieve klinische validatie. Figuur 2 geeft een overzicht op protocolniveau van deze workflow, waarin het gebruik van datasets, voorverwerking, feature-extractie, classifiertraining en evaluatiescenario's worden geïllustreerd. Terwijl Figuur 1 zich richt op de interne modelarchitectuur, legt Figuur 2 de nadruk op experimenteel ontwerp en datastroom over trainings- en testfasen.

Figuur 2: Workflow van het voorgestelde COVID-19 hoestscreeningsraamwerk. Illustratie van de volledige verwerkingspijplijn, inclusief preprocessing, feature-extractie (MFCC, chroma, spectrale kenmerken), GAN–VAE-gebaseerde representatieleer en -augmentatie (alleen training), en definitieve classificatie onder onderwerpniveau splitsing en cross-dataset-evaluatie. Klik hier om een grotere versie van deze figuur te bekijken.