Saved in:
Bibliographic Details
Main Authors: Wang, Yanghao, Chen, Long
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2408.16266
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916489879093248
author Wang, Yanghao
Chen, Long
author_facet Wang, Yanghao
Chen, Long
contents Data Augmentation (DA), i.e., synthesizing faithful and diverse samples to expand the original training set, is a prevalent and effective strategy to improve the performance of various data-scarce tasks. With the powerful image generation ability, diffusion-based DA has shown strong performance gains on different image classification benchmarks. In this paper, we analyze today's diffusion-based DA methods, and argue that they cannot take account of both faithfulness and diversity, which are two critical keys for generating high-quality samples and boosting classification performance. To this end, we propose a novel Diffusion-based DA method: Diff-II. Specifically, it consists of three steps: 1) Category concepts learning: Learning concept embeddings for each category. 2) Inversion interpolation: Calculating the inversion for each image, and conducting circle interpolation for two randomly sampled inversions from the same category. 3) Two-stage denoising: Using different prompts to generate synthesized images in a coarse-to-fine manner. Extensive experiments on various data-scarce image classification tasks (e.g., few-shot, long-tailed, and out-of-distribution classification) have demonstrated its effectiveness over state-of-the-art diffusion-based DA methods.
format Preprint
id arxiv_https___arxiv_org_abs_2408_16266
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Inversion Circle Interpolation: Diffusion-based Image Augmentation for Data-scarce Classification
Wang, Yanghao
Chen, Long
Computer Vision and Pattern Recognition
Data Augmentation (DA), i.e., synthesizing faithful and diverse samples to expand the original training set, is a prevalent and effective strategy to improve the performance of various data-scarce tasks. With the powerful image generation ability, diffusion-based DA has shown strong performance gains on different image classification benchmarks. In this paper, we analyze today's diffusion-based DA methods, and argue that they cannot take account of both faithfulness and diversity, which are two critical keys for generating high-quality samples and boosting classification performance. To this end, we propose a novel Diffusion-based DA method: Diff-II. Specifically, it consists of three steps: 1) Category concepts learning: Learning concept embeddings for each category. 2) Inversion interpolation: Calculating the inversion for each image, and conducting circle interpolation for two randomly sampled inversions from the same category. 3) Two-stage denoising: Using different prompts to generate synthesized images in a coarse-to-fine manner. Extensive experiments on various data-scarce image classification tasks (e.g., few-shot, long-tailed, and out-of-distribution classification) have demonstrated its effectiveness over state-of-the-art diffusion-based DA methods.
title Inversion Circle Interpolation: Diffusion-based Image Augmentation for Data-scarce Classification
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.16266