返回
Data augmentation for Gram-stain images based on Vector Quantized Variational AutoEncoder
DOI:10.1016/j.neucom.2024.128123.png)
摘要
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总结
对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。
期刊
IF:
6.5
论文数:
2.5W
被引数:
6.5W
机构
暂无机构信息
引用论文
Human-related anomalous event detection via memory-augmented Wasserstein generative adversarial network with gradient penalty通过具有梯度惩罚的记忆增强Wasserstein生成对抗网络检测与人类相关的异常事件
PATTERN RECOGNITION
IF7.6
Deep Convolutional Neural Networks for Computer-Aided Detection: CNN Architectures, Dataset Characteristics and Transfer Learning用于计算机辅助检测的深度卷积神经网络: CNN架构,数据集特征和迁移学习
Atomic-level Ru-Ir mixing in rutile-type (RuIr)O2 for efficient and durable oxygen evolution catalysis锐钛矿型(RuIr)O2中原子级别的Ru-Ir混合用于高效且耐用的氧析出催化
NATURE COMMUNICATIONS
IF15.7
Generation of whole building renovation scenarios using variational autoencoders
ENERGY AND BUILDINGS
IF7.1

