MGAug: Multimodal Geometric Augmentation in Latent Spaces of Image Deformations

2023-12-20 · Tonmoy Hossain, Miaomiao Zhang

Geometric transformations have been widely used to augment the size of training images. Existing methods often assume a unimodal distribution of the underlying transformations between images, which limits their power when data with multimodal distributions occur. In this paper, we propose a novel model, Multimodal Geometric Augmentation (MGAug), that for the first time generates augmenting transformations in a multimodal latent space of geometric deformations. To achieve this, we first develop a deep network that embeds the learning of latent geometric spaces of diffeomorphic transformations (a.k.a. diffeomorphisms) in a variational autoencoder (VAE). A mixture of multivariate Gaussians is formulated in the tangent space of diffeomorphisms and serves as a prior to approximate the hidden distribution of image transformations. We then augment the original training dataset by deforming images using randomly sampled transformations from the learned multimodal latent space of VAE. To validate the efficiency of our model, we jointly learn the augmentation strategy with two distinct domain-specific tasks: multi-class classification on 2D synthetic datasets and segmentation on real 3D brain magnetic resonance images (MRIs). We also compare MGAug with state-of-the-art transformation-based image augmentation algorithms. Experimental results show that our proposed approach outperforms all baselines by significantly improved prediction accuracy. Our code is publicly available at https://github.com/tonmoy-hossain/MGAug.

📄 PDF Abstract BibTeX arXiv:2312.13440

Code (1)

tonmoy-hossain/mgaug 공식 구현 pytorch

Tasks

Image Augmentation Multi-class Classification

Improving Generalization in Meta-Learning via Meta-Gradient Augmentation

2023-06-14 · Ren Wang, Haoliang Sun, Qi Wei, Xiushan Nie 외

Meta-learning methods typically follow a two-loop framework, where each loop potentially suffers from notorious overfitting, hindering rapid adaptation and generalization to new tasks. Existing schemes solve it by enhanc…

Few-Shot Learning Memorization Meta-Learning Network Pruning

Geometry of Uncertainty: Learning Metric Spaces for Multimodal State Estimation in RL

2026-02-12 · Alfredo Reichlin, Adriano Pacciarelli, Danica Kragic, Miguel Vasco arxiv

Estimating the state of an environment from high-dimensional, multimodal, and noisy observations is a fundamental challenge in reinforcement learning (RL). Traditional approaches rely on probabilistic models to account f…

Reinforcement Learning

SEMASIA: A Large-Scale Dataset of Semantically Structured Latent Representations

2026-05-10 · Mario Edoardo Pandolfo, Enrico Grimaldi, Lorenzo Marinucci, Leonardo Di Nino 외 arxiv

Latent representations learned by neural networks often exhibit semantic structure, where concept similarity is reflected by geometric proximity in embedding space. However, comparing such spaces across models remains di…

Semantic Communication Transfer Learning

Topological Obstructions and How to Avoid Them

2023-12-12 · NeurIPS 2023 11 · Babak Esmaeili, Robin Walters, Heiko Zimmermann, Jan-Willem van de Meent

Incorporating geometric inductive biases into models can aid interpretability and generalization, but encoding to a specific geometric structure can be challenging due to the imposed topological constraints. In this pape…

GLASS: Geometric Latent Augmentation for Shape Spaces

2021-08-06 · CVPR 2022 1 · Sanjeev Muralikrishnan, Siddhartha Chaudhuri, Noam Aigerman, Vladimir Kim 외

We investigate the problem of training generative models on a very sparse collection of 3D models. We use geometrically motivated energies to augment and thus boost a sparse collection of example (training) models. We an…

valid