$$\rightleftharpoonup{xx}$$
$$\longleftharp{xx}$$,
$$\longrightharp{xx}$$,
Metodologia Proposta
Este estudo propõe uma estrutura baseada em IA generativa para detectar a COVID-19 a partir de sinais de tosse. A estrutura integra modelos generativos com classificadores discriminativos, mantendo os dados de treinamento e avaliação estritamente separados. A Figura 1 ilustra a arquitetura detalhada do framework proposto GAN–VAE–ADCNN, mostrando o fluxo do pré-processamento de áudio e extração de características até o aprendizado de representações latentes via um Autoencoder Variacional (VAE), geração de características sintéticas usando uma Rede Generativa Adversarial (GAN), e a classificação final usando Redes Neurais Convolucionais Profundas (DCNN) e DCNN baseada em atenção (ADCNN). A figura mostra que componentes generativos são usados apenas durante o treinamento, enquanto a classificação é realizada usando modelos discriminativos.

Figura 1: Visão geral da arquitetura GAN–VAE–ADCNN. Esquema do pipeline híbrido proposto, onde características acústicas derivadas da tosse são codificadas usando um VAE, ampliadas por geração de amostras sintéticas baseadas em GAN e classificadas usando modelos DCNN e DCNN baseado em atenção (ADCNN). Por favor, clique aqui para ver uma versão ampliada desta figura.
Arquitetura e Implementação do Modelo
Os modelos generativos e discriminativos usados nesse framework foram implementados com arquiteturas fixas e reprodutíveis. O GAN consiste em um gerador com três camadas totalmente conectadas (128, 256 e 512 unidades) usando ativação ReLU, e um discriminador com três camadas totalmente conectadas (512, 256 e 128 unidades) usando ativação LeakyReLU seguida por uma saída sigmoide.
O codificador VAE é composto por duas camadas totalmente conexas com 256 e 128 unidades, seguidas por estimativas de média latente e variância com dimensionalidade latente de 64. O decodificador espelha essa estrutura e é treinado usando uma combinação de perda por reconstrução e divergência de Kullback–Leibler para aprender um espaço latente estruturado e probabilístico.
O classificador DCNN inclui quatro blocos convolucionais com kernels 3x3 e 32, 64, 128 e 256 filtros, respectivamente. Cada bloco é seguido por normalização em lote, ativação do ReLU e 2x2 max pooling. O ADCNN estende o DCNN incorporando um mecanismo de atenção canal a canal após o bloco convolucional final. Todos os modelos foram otimizados usando o otimizador Adam, com taxa de aprendizado de 0,0001 e tamanho de lote de 32. Como mostrado na Figura 1, o VAE e o GAN operam apenas durante o treinamento, enquanto DCNN e ADCNN são usados para classificação. O procedimento completo de treinamento e avaliação é resumido no Algoritmo 1.
Algoritmo 1: Estrutura de Detecção de COVID-19 Baseada em GAN e VAE
Entrada: Gravações de áudio pré-processadas para a tosse
Saída: Rótulo de classificação binária (COVID-19 positivo/negativo)
Os procedimentos de treinamento e avaliação seguiram um pipeline fixo e reproduzível. Todas as gravações de áudio da tosse foram reamostradas para 16 kHz, submetidas a redução de ruído e normalizadas em amplitude. As gravações foram segmentadas em segmentos de comprimento fixo, dos quais foram extraídas MFCC, chroma e características espectrais. Um total de 40 Coeficientes Cepstrais de Mel-Frequência (MFCCs) foram extraídos de cada segmento de áudio. Além disso, 12 características de cromina foram calculadas. A representação resultante forma um vetor de características de 52 dimensões para cada segmento de tosse. Foi realizada a divisão em nível de sujeito para dividir os dados em conjuntos de treinamento, validação e teste. O VAE foi treinado para aprender representações latentes probabilísticas, e os vetores latentes resultantes foram usados para treinar o GAN para geração de características sintéticas. O conjunto de dados de treinamento foi ampliado usando amostras geradas pelo GAN–VAE, enquanto dados sintéticos foram excluídos da validação e dos testes. Classificadores DCNN e ADCNN foram treinados com os dados de treinamento aumentados, e a avaliação final foi realizada no conjunto de testes estendido usando métricas padrão de desempenho.
Configuração de Treinamento, Divisão de Dados e Prevenção de Vazamentos
Todos os experimentos foram conduzidos com uma configuração de treinamento fixa e reprodutível. A divisão de dados era realizada no nível do sujeito antes da segmentação de áudio para evitar vazamento de dados. O conjunto de dados foi dividido em conjuntos de treinamento, validação e teste em uma proporção de 80:10:10, garantindo que todos os segmentos da mesma gravação fossem atribuídos a um único subconjunto. O conjunto de treinamento era usado para aprendizado de modelos e aumento de dados generativos, o conjunto de validação para ajuste de hiperparâmetros e parada precoce, e o conjunto de teste era mantido para avaliação final de desempenho. Amostras sintéticas geradas pelo framework GAN–VAE foram usadas exclusivamente durante o treinamento e foram estritamente excluídas da validação e dos testes para garantir uma avaliação justa.
Os modelos foram treinados por um máximo de 100 épocas, com paradas precoces baseadas na perda de validação e uma paciência de 10 épocas. A otimização foi realizada usando o otimizador Adam com taxa de aprendizado de 0,0001 e tamanho de lote de 32. A perda binária de entropia cruzada foi aplicada para classificação, enquanto perdas de reconstrução e adversariais foram usadas para treinar os componentes VAE e GAN, respectivamente. Esse protocolo unificado de treinamento e avaliação garante reprodutibilidade, previne vazamento de dados e mantém uma separação rigorosa entre as etapas de treinamento e avaliação.
Descrição do Conjunto de Dados
Dois conjuntos de dados de áudio para tosse disponíveis publicamente — COUGHVID e Virufy — foram usados para equilibrar a diversidade acústica em grande escala com rótulos clinicamente referenciados, com papéis claramente separados no treinamento e avaliação.
COUGHVID é uma coletânea colaborativa de gravações de tosse, com anotação parcial de especialistas e metadados autorrelatados. Para garantir a qualidade dos dados, 428 gravações foram selecionadas após aplicação de critérios pré-definidos de controle de qualidade, incluindo duração mínima do sinal, relação sinal-ruído adequada, remoção de amostras corrompidas ou ambíguas e presença de eventos identificáveis de tosse com metadados de sintomas. Após pré-processamento e segmentação, essas gravações renderam 1.719 segmentos de tosse, padronizados para o formato WAV com uma taxa de amostragem de 16 kHz. O COUGHVID foi usado para treinar tanto os modelos generativos quanto os classificadores discriminativos.
O Virufy consiste em gravações de tosse supervisionadas por médicos, rotuladas como RT-PCR positiva ou negativa para COVID-19. Todas as 16 gravações disponíveis foram incluídas, resultando em 234 segmentos de tosse após segmentação, também padronizados para 16 kHz. O Virufy foi usado exclusivamente para avaliação externa de conjuntos de dados cruzados para avaliar o desempenho de generalização e não foi usado para treinamento, ajuste fino ou aumento generativo.
O arcabouço proposto deve, portanto, ser interpretado como uma abordagem de triagem de prova de conceito avaliada sob condições experimentais controladas, em vez de um sistema diagnóstico clinicamente validado.
Pré-processamento e segmentação de dados
Todas as gravações foram reamostradas para 16 kHz, convertidas para mono e submetidas à remoção de silêncio, redução de ruído espectral e normalização de amplitude. A segmentação era realizada após a divisão em nível de sujeito para evitar vazamento de dados. Cada gravação era dividida em segmentos sobrepostos de 2 a 4 s, e segmentos de baixa energia ou silenciosos eram descartados.
Protocolo de Validação Externa
A validação externa foi realizada por meio de avaliação cruzada de conjuntos de dados. Modelos treinados com COUGHVID foram avaliados no Virufy para validação externa. Esse protocolo avalia a generalização entre conjuntos de dados coletados sob condições diferentes, em vez da validação clínica prospectiva. A Figura 2 oferece uma visão geral em nível de protocolo desse fluxo de trabalho, ilustrando o uso do conjunto de dados, pré-processamento, extração de características, treinamento de classificadores e cenários de avaliação. Enquanto a Figura 1 foca na arquitetura interna do modelo, a Figura 2 enfatiza o projeto experimental e o fluxo de dados entre as etapas de treinamento e teste.

Figura 2: Fluxo de trabalho do framework proposto para rastreamento de tosse COVID-19. Ilustração de todo o pipeline de processamento, incluindo pré-processamento, extração de características (MFCC, croma, características espectrais), aprendizado e aumento de representações baseados em GAN–VAE (apenas treinamento), e classificação final sob divisão em nível de sujeito e avaliação cruzada de conjuntos de dados. Por favor, clique aqui para ver uma versão ampliada desta figura.