Fit Pixels, Get Labels: Meta-learned Implicit Networks for Image Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Vyas, Kushal, Veeraraghavan, Ashok, Balakrishnan, Guha
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918154594156544
author Vyas, Kushal
Veeraraghavan, Ashok
Balakrishnan, Guha
author_facet Vyas, Kushal
Veeraraghavan, Ashok
Balakrishnan, Guha
contents Implicit neural representations (INRs) have achieved remarkable successes in learning expressive yet compact signal representations. However, they are not naturally amenable to predictive tasks such as segmentation, where they must learn semantic structures over a distribution of signals. In this study, we introduce MetaSeg, a meta-learning framework to train INRs for medical image segmentation. MetaSeg uses an underlying INR that simultaneously predicts per pixel intensity values and class labels. It then uses a meta-learning procedure to find optimal initial parameters for this INR over a training dataset of images and segmentation maps, such that the INR can simply be fine-tuned to fit pixels of an unseen test image, and automatically decode its class labels. We evaluated MetaSeg on 2D and 3D brain MRI segmentation tasks and report Dice scores comparable to commonly used U-Net models, but with $90\%$ fewer parameters. MetaSeg offers a fresh, scalable alternative to traditional resource-heavy architectures such as U-Nets and vision transformers for medical image segmentation. Our project is available at https://kushalvyas.github.io/metaseg.html .
format Preprint
id arxiv_https___arxiv_org_abs_2510_04021
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Fit Pixels, Get Labels: Meta-learned Implicit Networks for Image Segmentation
Vyas, Kushal
Veeraraghavan, Ashok
Balakrishnan, Guha
Computer Vision and Pattern Recognition
Implicit neural representations (INRs) have achieved remarkable successes in learning expressive yet compact signal representations. However, they are not naturally amenable to predictive tasks such as segmentation, where they must learn semantic structures over a distribution of signals. In this study, we introduce MetaSeg, a meta-learning framework to train INRs for medical image segmentation. MetaSeg uses an underlying INR that simultaneously predicts per pixel intensity values and class labels. It then uses a meta-learning procedure to find optimal initial parameters for this INR over a training dataset of images and segmentation maps, such that the INR can simply be fine-tuned to fit pixels of an unseen test image, and automatically decode its class labels. We evaluated MetaSeg on 2D and 3D brain MRI segmentation tasks and report Dice scores comparable to commonly used U-Net models, but with $90\%$ fewer parameters. MetaSeg offers a fresh, scalable alternative to traditional resource-heavy architectures such as U-Nets and vision transformers for medical image segmentation. Our project is available at https://kushalvyas.github.io/metaseg.html .
title Fit Pixels, Get Labels: Meta-learned Implicit Networks for Image Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2510.04021