Search papers, labs, and topics across Lattice.
This paper introduces DEQ-Unmix, a deep equilibrium network for hyperspectral unmixing that reformulates abundance estimation as a fixed-point problem solved via implicit differentiation. The model replaces the data reconstruction term's gradient operator with a trainable convolutional network to better capture spectral-spatial information. Experiments on synthetic and real-world datasets show DEQ-Unmix achieves state-of-the-art unmixing performance with constant memory cost during training.
Hyperspectral unmixing just got a memory-efficient deep learning upgrade, achieving state-of-the-art performance without the backpropagation bloat.
Hyperspectral unmixing (HU) is crucial for analyzing hyperspectral imagery, yet achieving accurate unmixing remains challenging. While traditional methods struggle to effectively model complex spectral-spatial features, deep learning approaches often lack physical interpretability. Unrolling-based methods, despite offering network interpretability, inadequately exploit spectral-spatial information and incur high memory costs and numerical precision issues during backpropagation. To address these limitations, we propose DEQ-Unmix, which reformulates abundance estimation as a deep equilibrium model, enabling efficient constant-memory training via implicit differentiation. It replaces the gradient operator of the data reconstruction term with a trainable convolutional network to capture spectral-spatial information. By leveraging implicit differentiation, DEQ-Unmix enables efficient and constant-memory backpropagation. Experiments on synthetic and two real-world datasets demonstrate that DEQ-Unmix achieves superior unmixing performance while maintaining constant memory cost.
