arrow
返回

Data augmentation for Gram-stain images based on Vector Quantized Variational AutoEncoder

delete2024-10-01
delete2
delete
OA
AI
S
Shwetha, V
K
Keerthana Prasad *
C
Chiranjay Mukhopadhyay
B
Barnini Banerjee
DOI:10.1016/j.neucom.2024.128123delete
delete原文链接
delete分享
delete收藏
查看原文
摘要

摘要

En 中文
Availability of large-scale datasets plays a significant role in segmentation and classification tasks using deep learning. However, domains such as healthcare inherently suffer from unavailability and inaccessibility of data. This leads to challenges in deploying CNN-based models for computer-aided diagnosis. This challenge extends to Gram-stain image analysis for detecting bacterial infections, which is a crucial task. The lack of datasets containing Gram-stained direct and culture smear images exacerbates the significant challenges in deep learning tasks. In this regard, we investigate a novel application of the Variational AutoEncoder. Specifically, the Vector Quantized Variational AutoEncoder model is trained to generate the Gram-stain images. Incorporating a novel loss function, where the quality loss ( L qu ) is derived by integrating the Loss SSIM , L 1 , and L 2 losses with the VQ-VAE loss ( Loss vq ) for proposed approach for Gram-stained direct and culture smear images. This modification facilitates the creation of images closely resembling the original input, leading to notable SSIM scores of 0.92 for Gram-stained culture images and 0.88 for Gram-stained direct smear images. The current study compares the proposed method with state-of-the-art machine learning based and CNN based transformations. This work also demonstrates the classification process with and without image augmentation. It shows that the area under the curve in the case of augmentation is higher by an average of 20%.
Keyword:
Bacteria detection
Generative Adversarial network
Variational AutoEncoder
VQ-VAE
Augmentation
Transformation
Semantic segmentation
CNN based classification
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

Neurocomputing 封面图
Neurocomputing
IF:
6.5
论文数:
2.5W
被引数:
6.5W

机构

暂无机构信息
引用论文

引用论文

Auto-encoder based dimensionality reduction基于自动编码器的降维
err2016-04-01
err675
PREAI
errWang, Yasi; Yao, Hongxun; Zhao, Sicheng
err分享
err收藏
Generation of whole building renovation scenarios using variational autoencoders
err2021-01-01
err21
PREAI
errSharif, Seyed Amirhosain; Hammad, Amin; Eshraghi, Pegah
err分享
err收藏
Selektive RGD-vermittelte Adhäsion von Osteoblasten an Implantat-Oberflächen
err1999-02-15
err0
PREAI
errMartin Kantlehner; Dirk Finsinger; Jörg Meyer; Patricia Schaffner; Alfred Jonczyk; Beate Diefenbach; Berthold Nies; Horst Kessler
err分享
err收藏
err分享
err收藏
Tunable Raman gain in mid-IR waveguides
err2017-12-12
err0
errOAAI
errA. D. Sánchez; S. M. Hernandez; J. Bonetti; P. I. Fierens; D. F. Grosz
err分享
err收藏
学者 查看更多内容