LoFi: Neural Local Fields for Scalable Image Reconstruction

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Khorashadizadeh, AmirEhsan, Liaudat, Tobías I., Liu, Tianlin, McEwen, Jason D., Dokmanić, Ivan
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912165000118272
author Khorashadizadeh, AmirEhsan
Liaudat, Tobías I.
Liu, Tianlin
McEwen, Jason D.
Dokmanić, Ivan
author_facet Khorashadizadeh, AmirEhsan
Liaudat, Tobías I.
Liu, Tianlin
McEwen, Jason D.
Dokmanić, Ivan
contents Neural fields or implicit neural representations (INRs) have attracted significant attention in computer vision and imaging due to their efficient coordinate-based representation of images and 3D volumes. In this work, we introduce a coordinate-based framework for solving imaging inverse problems, termed LoFi (Local Field). Unlike conventional methods for image reconstruction, LoFi processes local information at each coordinate separately by multi-layer perceptrons (MLPs), recovering the object at that specific coordinate. Similar to INRs, LoFi can recover images at any continuous coordinate, enabling image reconstruction at multiple resolutions. With comparable or better performance than standard deep learning models like convolutional neural networks (CNNs) and vision transformers (ViTs), LoFi achieves excellent generalization to out-of-distribution data with memory usage almost independent of image resolution. Remarkably, training on 1024x1024 images requires less than 200MB of memory -- much below standard CNNs and ViTs. Additionally, LoFi's local design allows it to train on extremely small datasets with 10 samples or fewer, without overfitting and without the need for explicit regularization or early stopping.
format Preprint
id arxiv_https___arxiv_org_abs_2411_04995
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle LoFi: Neural Local Fields for Scalable Image Reconstruction
Khorashadizadeh, AmirEhsan
Liaudat, Tobías I.
Liu, Tianlin
McEwen, Jason D.
Dokmanić, Ivan
Computer Vision and Pattern Recognition
Machine Learning
Neural fields or implicit neural representations (INRs) have attracted significant attention in computer vision and imaging due to their efficient coordinate-based representation of images and 3D volumes. In this work, we introduce a coordinate-based framework for solving imaging inverse problems, termed LoFi (Local Field). Unlike conventional methods for image reconstruction, LoFi processes local information at each coordinate separately by multi-layer perceptrons (MLPs), recovering the object at that specific coordinate. Similar to INRs, LoFi can recover images at any continuous coordinate, enabling image reconstruction at multiple resolutions. With comparable or better performance than standard deep learning models like convolutional neural networks (CNNs) and vision transformers (ViTs), LoFi achieves excellent generalization to out-of-distribution data with memory usage almost independent of image resolution. Remarkably, training on 1024x1024 images requires less than 200MB of memory -- much below standard CNNs and ViTs. Additionally, LoFi's local design allows it to train on extremely small datasets with 10 samples or fewer, without overfitting and without the need for explicit regularization or early stopping.
title LoFi: Neural Local Fields for Scalable Image Reconstruction
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2411.04995