For spatial-spectral classification of hyperspectral images (HSI), a deep learning framework is proposed in this paper, which consists of convolutional neural networks (CNN) and Markov random fields (MRF). Firstly, a CNN model to learn the deep spectral feature from the HSI is built and the class posterior probability distribution is estimated. The CNN with a dropout layer can relieve the overfitting in classification. The CNN is utilized as a pixel-classifier, so it only works in the spectral domain. Then, the spatial information will be encoded by MRF-based multilevel logistic (MLL) prior for regularizing the classification. To derive the correlation of both spectral and spatial features for improving algorithm performance, the marginal probability distribution in HSI is learned using MRF-based loopy belief propagation (LBP). In comparison with several state-of-the-art approaches for data classification on 3 publicly available HSI datasets, experimental results have demonstrated the superior performance of the proposed methodology.
- hyperspectral image
- spatial-spectral classification
- convolutional neural networks
- Markov random fields
- loopy belief propagation