kth.sePublications KTH
Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
A New Hyperspectral Reconstruction Method with Conditional Diffusion Model for Snapshot Spectral Compressive Imaging
Zhejiang University, Centre for Optical and Electromagnetic Research, National Engineering Research Center for Optical Instruments, Zhejiang Provincial Key Laboratory for Sensing Technologies, Hangzhou, China, 310058.
Zhejiang University, Centre for Optical and Electromagnetic Research, National Engineering Research Center for Optical Instruments, Zhejiang Provincial Key Laboratory for Sensing Technologies, Hangzhou, China, 310058; Shanghai Institute for Advanced Study, Zhejiang University, Shanghai, China, 201203.
Westlake University, School of Engineering, Hangzhou, China, 310030.
KTH, School of Electrical Engineering and Computer Science (EECS), Electrical Engineering, Electromagnetic Engineering and Fusion Science. Zhejiang University, Centre for Optical and Electromagnetic Research, National Engineering Research Center for Optical Instruments, Zhejiang Provincial Key Laboratory for Sensing Technologies, Hangzhou, China, 310058.ORCID iD: 0000-0002-3401-1125
2025 (English)In: IEEE Transactions on Instrumentation and Measurement, ISSN 0018-9456, E-ISSN 1557-9662, Vol. 74, article id 4506214Article in journal (Refereed) Published
Abstract [en]

In the coded aperture snapshot spectral imaging (CASSI) system, the coded and compressed single-channel measurements need to be reconstructed into hyperspectral cubes. Existing discriminative models reconstruct the spectral cube by optimizing the mean squared error (MSE) between the ground truth and the predicted image, employing peak signal-to-noise ratio (PSNR) and structural similarity index (SSIM) as metrics to gauge the quality of reconstruction. However, these indicators often possess significant limitations in mimicking human visual perception and in discerning the impact of image distortions on perceived visual quality. In this article, a new model named CASSIDiff is proposed to reconstruct CASSI measurements, achieving advanced results in perceptual loss-based evaluation metrics such as learned perceptual image patch similarity (LPIPS) and Fréchet inception distance (FID). The diffusion model, which enjoys high accuracy and reliability in generative tasks, is used for the first time for the hyperspectral reconstruction task. A feature fusion mechanism based on discrete wavelet transform (DWT) is used to weaken the noise interference effect in the conditional diffusion model. Considering the interspectra similarity and long-range dependencies of hyperspectral data, the spatial-spectral attention mechanism is also introduced. Experiments show that CASSIDiff not only outperforms most existing algorithms in simulation datasets but also shows robustness to real data published and collected in our home-built CASSI system. The code and models are publicly available at: https://github.com/YifanSi/CASSIDiff.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE) , 2025. Vol. 74, article id 4506214
Keywords [en]
Coded aperture snapshot spectral imaging (CASSI), conditional diffusion model, Frchet inception distance (FID), hyperspectral reconstruction, learned perceptual image patch similarity (LPIPS)
National Category
Computer graphics and computer vision Signal Processing Computer Sciences
Identifiers
URN: urn:nbn:se:kth:diva-362703DOI: 10.1109/TIM.2025.3551465ISI: 001457758700032Scopus ID: 2-s2.0-105002391455OAI: oai:DiVA.org:kth-362703DiVA, id: diva2:1954145
Note

QC 20250520

Available from: 2025-04-23 Created: 2025-04-23 Last updated: 2025-05-20Bibliographically approved

Open Access in DiVA

No full text in DiVA

Other links

Publisher's full textScopus

Authority records

He, Sailing

Search in DiVA

By author/editor
He, Sailing
By organisation
Electromagnetic Engineering and Fusion Science
In the same journal
IEEE Transactions on Instrumentation and Measurement
Computer graphics and computer visionSignal ProcessingComputer Sciences

Search outside of DiVA

GoogleGoogle Scholar

doi
urn-nbn

Altmetric score

doi
urn-nbn
Total: 145 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf