Generative Active Learning with Variational Autoencoder for Radiology Data Generation in Veterinary Medicine

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lee, In-Gyu, Oh, Jun-Young, Yu, Hee-Jung, Kim, Jae-Hwan, Eom, Ki-Dong, Jeong, Ji-Hoon
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910355349831680
author Lee, In-Gyu
Oh, Jun-Young
Yu, Hee-Jung
Kim, Jae-Hwan
Eom, Ki-Dong
Jeong, Ji-Hoon
author_facet Lee, In-Gyu
Oh, Jun-Young
Yu, Hee-Jung
Kim, Jae-Hwan
Eom, Ki-Dong
Jeong, Ji-Hoon
contents Recently, with increasing interest in pet healthcare, the demand for computer-aided diagnosis (CAD) systems in veterinary medicine has increased. The development of veterinary CAD has stagnated due to a lack of sufficient radiology data. To overcome the challenge, we propose a generative active learning framework based on a variational autoencoder. This approach aims to alleviate the scarcity of reliable data for CAD systems in veterinary medicine. This study utilizes datasets comprising cardiomegaly radiograph data. After removing annotations and standardizing images, we employed a framework for data augmentation, which consists of a data generation phase and a query phase for filtering the generated data. The experimental results revealed that as the data generated through this framework was added to the training data of the generative model, the frechet inception distance consistently decreased from 84.14 to 50.75 on the radiograph. Subsequently, when the generated data were incorporated into the training of the classification model, the false positive of the confusion matrix also improved from 0.16 to 0.66 on the radiograph. The proposed framework has the potential to address the challenges of data scarcity in medical CAD, contributing to its advancement.
format Preprint
id arxiv_https___arxiv_org_abs_2403_03642
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Generative Active Learning with Variational Autoencoder for Radiology Data Generation in Veterinary Medicine
Lee, In-Gyu
Oh, Jun-Young
Yu, Hee-Jung
Kim, Jae-Hwan
Eom, Ki-Dong
Jeong, Ji-Hoon
Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
Recently, with increasing interest in pet healthcare, the demand for computer-aided diagnosis (CAD) systems in veterinary medicine has increased. The development of veterinary CAD has stagnated due to a lack of sufficient radiology data. To overcome the challenge, we propose a generative active learning framework based on a variational autoencoder. This approach aims to alleviate the scarcity of reliable data for CAD systems in veterinary medicine. This study utilizes datasets comprising cardiomegaly radiograph data. After removing annotations and standardizing images, we employed a framework for data augmentation, which consists of a data generation phase and a query phase for filtering the generated data. The experimental results revealed that as the data generated through this framework was added to the training data of the generative model, the frechet inception distance consistently decreased from 84.14 to 50.75 on the radiograph. Subsequently, when the generated data were incorporated into the training of the classification model, the false positive of the confusion matrix also improved from 0.16 to 0.66 on the radiograph. The proposed framework has the potential to address the challenges of data scarcity in medical CAD, contributing to its advancement.
title Generative Active Learning with Variational Autoencoder for Radiology Data Generation in Veterinary Medicine
topic Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2403.03642