Beyond Subspace Isolation: Many-to-Many Transformer for Light Field Image Super-resolution
About
The effective extraction of spatial-angular features plays a crucial role in light field image super-resolution (LFSR) tasks, and the introduction of convolution and Transformers leads to significant improvement in this area. Nevertheless, due to the large 4D data volume of light field images, many existing methods opted to decompose the data into a number of lower-dimensional subspaces and perform Transformers in each sub-space individually. As a side effect, these methods inadvertently restrict the self-attention mechanisms to a One-to-One scheme accessing only a limited subset of LF data, explicitly preventing comprehensive optimization on all spatial and angular cues. In this paper, we identify this limitation as subspace isolation and introduce a novel Many-to-Many Transformer (M2MT) to address it. M2MT aggregates angular information in the spatial subspace before performing the self-attention mechanism. It enables complete access to all information across all sub-aperture images (SAIs) in a light field image. Consequently, M2MT is enabled to comprehensively capture long-range correlation dependencies. With M2MT as the foundational component, we develop a simple yet effective M2MT network for LFSR. Our experimental results demonstrate that M2MT achieves state-of-the-art performance across various public datasets, and it offers a favorable balance between model performance and efficiency, yielding higher-quality LFSR results with substantially lower demand for memory and computation. We further conduct in-depth analysis using local attribution maps (LAM) to obtain visual interpretability, and the results validate that M2MT is empowered with a truly non-local context in both spatial and angular subspaces to mitigate subspace isolation and acquire effective spatial-angular representation.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Light Field Super-Resolution | INRIA 7x7 SAIs (test) | PSNR31.99 | 13 | |
| Light Field Super-Resolution | HCIold 7x7 SAIs (test) | PSNR38.22 | 13 | |
| Light Field Super-Resolution | STFgantry 7x7 SAIs (test) | PSNR32.68 | 13 | |
| Light Field Super-Resolution | EPFL 7x7 SAIs (test) | PSNR30.14 | 13 | |
| Light Field Super-Resolution | HCInew 7x7 SAIs (test) | PSNR31.8 | 13 | |
| Light Field Image Super-Resolution | HCInew 2x scale (test) | PSNR38.43 | 11 | |
| Light Field Image Super-Resolution | HCIold 2x scale (test) | PSNR45.38 | 11 | |
| Light Field Image Super-Resolution | INRIA 2x scale (test) | PSNR37.22 | 11 | |
| Light Field Image Super-Resolution | Average 2x scale (test) | PSNR39.53 | 11 | |
| Light Field Image Super-Resolution | EPFL 4x scale (test) | PSNR29.85 | 11 |