• Home
  • Search
  • PREmbed: Balancing Conditional Generative Models with Embedding Pretraining and Regularization
  • Cite Icon1
  • https://doi.org/10.3390/electronics14020280Copy DOI Icon

PREmbed: Balancing Conditional Generative Models with Embedding Pretraining and Regularization

Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Recent advancements in conditional generative models, e.g., Conditional Variational AutoEncoder (CVAE) and Conditional Denoising Diffusion Probabilistic Model (CDDPM), which utilize class-specific embeddings for generating images in specific classes, have demonstrated exceptional capabilities in producing high-quality images on balanced datasets. However, their performance decreases with imbalanced real-world data, affecting the fidelity and diversity of the generated images. This paper discusses the reasons and solutions for the imbalance issue in CVAE and CDDPM. By selectively reweighting the gradients of the embedding layer and main network, we identify the embedding layer’s bias as a critical bottleneck in these models. Motivated by this, we propose Embedding Pretraining and Regularization (PREmbed), a training methodology for addressing biased learning in the embedding layer of these models. PREmbed consists of two key components: (i) Masked Autoencoder pretraining on the original imbalanced dataset to obtain a robust initial embedding, and (ii) an embedding regularization loss to preserve class-level distances during the training phase to improve imbalanced embedding learning. Implementing PREmbed leads to the enhanced convergence of a robust embedding layer. Our experiments on imbalanced CIFAR-10 and CUB-200 datasets demonstrate that PREmbed enhances the performance of CVAE and CDDPM; it outperforms the baseline CVAE by 22.3% (FID 18.63) and the CDDPM by 30.3% (FID 12.70) in imbalanced CIFAR-10; 33.7% (FID 24.08) and 26.5% (FID 13.90) in CUB-200; and 16.4% (FID 49.41) and 15.2% (FID 33.05) in ImageNet-LT (which is where it achieves the state-of-the-art performance in imbalanced conditional image generation task).

Similar Papers
  • Research Article
  • Citations58

A conditional variational autoencoding generative adversarial networks with self-modulation for rolling bearing fault diagnosis

  • Feb 14, 2022
  • Measurement
  • Yunpeng Liu +4
  • PDF
  • Research Article
  • Citations13

Performance Evaluation of Sentiment Analysis on Balanced and Imbalanced Dataset Using Ensemble Approach

  • May 05, 2022
  • Indian Journal of Science and Technology
  • Shini George +1
  • Research Article
  • Citations36

ECG signal generation based on conditional generative models

  • Jan 23, 2023
  • Biomedical Signal Processing and Control
  • Yong Xia +2
  • Book Chapter
  • Citations4

Minority–Majority Mix mean Oversampling Technique: An Efficient Technique to Improve Classification of Imbalanced Data Sets

  • Oct 17, 2019
  • Sachin Patil +1
  • Book Chapter
  • Citations3

Early Detection of Diabetic Retinopathy Using Multimodal Approach

  • Jan 01, 2022
  • S G Shaila +4
  • PDF
  • Research Article
  • Citations4

A Novel Classification Method Based on a Two-Phase Technique for Learning Imbalanced Text Data

  • Mar 13, 2022
  • Symmetry
  • Der-Chiang Li +3
  • Research Article
  • Citations1

LungNet-ViT: Efficient lung disease classification using a multistage vision transformer model from chest radiographs.

  • Mar 28, 2025
  • Journal of X-ray science and technology
  • V Padmavathi +1
  • Research Article
  • Citations8

A Hybrid Software Defects Prediction Model for Imbalance Datasets Using Machine Learning Techniques: (S-SVM Model)

  • Jun 16, 2023
  • Journal of Autonomous Intelligence
  • Mohd Mustaqeem +1
  • Research Article
  • Citations108

A synthetic informative minority over-sampling (SIMO) algorithm leveraging support vector machine to enhance learning from imbalanced datasets

  • Nov 29, 2017
  • Decision Support Systems
  • Saeed Piri +2
  • Conference Article
  • Citations111

Combining Over-Sampling and Under-Sampling Techniques for Imbalance Dataset

  • Feb 24, 2017
  • Nutthaporn Junsomboon +1
  • Research Article
  • Citations1

GANsemble for Small and Imbalanced Data Sets : A Baseline for Synthetic Microplastics Data

  • May 27, 2024
  • Proceedings of the Canadian Conference on Artificial Intelligence
  • Daniel Platnick +3
  • Research Article
  • Citations1

Evaluation of Machine Learning Models for Sentiment Analysis in the South Sumatra Governor Election Using Data Balancing Techniques

  • Mar 21, 2025
  • Journal of Information Systems and Informatics
  • Febriyanti Panjaitan +5
  • Conference Article
  • Citations5

Fault Diagnosis Method of Analog Circuit Based on Enhanced Boundary Equilibrium Generative Adversarial Networks

  • Oct 15, 2021
  • 2021 Global Reliability and Prognostics and Health Management (PHM-Nanjing)
  • Jingli Yang +3
  • PDF
  • Research Article
  • Citations196

Impact of SMOTE on Imbalanced Text Features for Toxic Comments Classification Using RVVC Model

  • Jan 01, 2021
  • IEEE Access
  • Vaibhav Rupapara +5
  • Research Article
  • Citations14

Identifying Heat Shock Protein Families from Imbalanced Data by Using Combined Features.

  • Sep 23, 2020
  • Computational and Mathematical Methods in Medicine
  • Xiao-Yang Jing +1
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.