The present disclosure relates to the technical field of hyperspectral data processing and graph signal processing, and in particularly, to a hyperspectral image distributed restoration method and system based on graph signal processing and superpixel segmentation.
A hyperspectral image (HSI) is composed of dozens or even hundreds of gray images corresponding to spectral channels, and has a high spectral resolution and rich texture information. Based on HSI features, a combination of a spatial image and a spectral band can provide favorable conditions for the extraction of target features. Therefore, the HSI is widely used in food quality and safety assessment, medical diagnosis and image-guided surgery, precision agriculture and other fields. However, the HSI is often disturbed by various noises in a process of acquisition and transmission. These noises bring great challenges to a subsequent analysis, especially in feature extraction and target detection, which will greatly reduce the reliability of processing. Therefore, how to effectively remove the noises of the HSI has become a key issue.
At present, various denoising methods have been proposed to restore the HSI. The traditional methods of HSI denoising include: regarding each band as a gray image, and applying a two-dimensional denoising method such as a block matching and 3D collaborative filtering (BM3D) algorithm to remove noise band by band. However, these methods ignore high correlation between spectral bands or spatial pixels, resulting in relatively poor restoration performance. In view of this, various HSI denoising methods have been proposed to consider the correlation, for example, a hybrid spatial-spectral noise reduction (HS SNR) method, where a wavelet shrinkage model in the derivative domain of mixed spatial spectrum was proposed, which considers the difference of signal regularity in spatial and spectral dimensions; color total variation (CTV) and spectral spatial adaptive hyperspectral total variation (SSAHTV), which apply an image operator called total variation (TV) to HSI restoration, the image operator describes a spatial piecewise smoothness (PWS) characteristic of the HSI; low-rank matrix recovery (LRMR), which uses a low-dimensional structure in high-dimensional data to remove noise based on low-rank (LR) matrix approximation; tensor subspace low-rank learning with non-local prior for hyperspectral image mixed denoising (TSLRLN) and total variation regularized low-rank tensor decomposition (LRTDTV), which utilize LR characteristic of the HSI based on tensor decomposition; Global and Nonlocal Low-Rank Factorizations (GLF), which combines different LR characteristics, etc. These methods effectively represent the spatial and spectral correlation of the HSI. In addition, there are some nonlocal methods, such as Nonlocal Tensor-Ring Decomposition (NLTR) and non-local meets global (NGMeet), which denoise the HSI group by group, but these methods only consider nonlocal self-similarity and ignore PWS characteristics in groups. Total Variation Regularization and Nonlocal Low-Rank Tensor Decomposition (TVNLRTD) regards a non-local self-similarity as a regularization term in a model, and combines the regularization term with other global regularization terms (such as TV), in which good restoration performance is achieved through combining the regularization term with the other global regularization terms, but it cannot simply restore the HSI group by group, and a disadvantage of high complexity of single computing node in large-scale HSI is brought out. Different from the non-local-based method, a superpixel segmentation-based method, such as superpixel segmentation of smooth band (SSSB) groups pixels of the HSI into a series of uniform regions, a shape of each region depends on boundary information, and the shape is irregular. Therefore, the method superpixel segmentation-based method can effectively preserve the boundary information, while the non-local-based method cannot preserve the boundary information because of square patch-based strategy. However, the existing superpixel segmentation-based methods, such as SSSB, do not perform well in HSI restoration, partly because they do not consider the PWS characteristics of the HSI and the similarity between superpixels. In view of the shortcomings of the above methods, graph signal processing (GSP) is considered, which is a multi-disciplinary research field, uses graph topology to describe the correlation between network data, comprehensively constructs the theoretical framework of network data processing, and greatly expands the theoretical boundary of classical signal processing. Therefore, a hyperspectral mixed noise removal method based on GSP and superpixel segmentation can be proposed.
Moreover, the existing methods usually focus on a small-scale HSI or a subregion of large-scale HSI and they exhibit poor scalability on large HSI since they employ the centralized operators in general. Besides, several restoration methods transform the three-dimensional HSI dataset into a two-dimensional matrix, an ill-posed matrix where the spatial dimension greatly exceeds the spectral one, resulting in blurring and a loss of details in denoised results. For large-scale HSI, the loss of boundary information and texture details will be more serious.
In addition, the existing HSI is often disturbed by various noises, which brings great challenges to the subsequent applications. Many existing restoration methods are not suitable for large-scale HSI, so how to improve the scalability of the restoration methods is a key issue.
In view of this, objectives of the present disclosure are to provide a large-scale hyperspectral image (HSI) distributed restoration method based on graph signal processing and superpixel segmentation.
To achieve the above objectives, the present disclosure provides the following technical solutions.
A large-scale hyperspectral image distributed restoration method based on graph signal processing and superpixel segmentation is provided according to an embodiment of the present disclosure, which includes:
step 1, constructing an input signal model, to linearly normalize a hyperspectral image to obtain a normalized hyperspectral image;
step 2, selecting an image corresponding to a target-band from the normalized hyperspectral image, and pre-denoising the image to obtain a pre-denoised image;
step 3, segmenting the pre-denoised image into multiple superpixel regions using simple linear iterative cluster (SLIC) superpixel segmentation algorithm;
step 4, constructing a skeleton graph based on the multiple superpixel regions;
step 5, constructing a local graph for all pixels of each of the multiple superpixel regions, to obtain multiple local graphs respectively corresponding to the multiple superpixel regions;
step 6, constructing sub-graphs based on the skeleton graph and the multiple local graphs, each of the sub-graphs being corresponding to a computing node;
step 7, applying an exchange distributed mode to each of the sub-graphs, to perform distributed denoising on the hyperspectral image;
step 8, establishing an optimization model of each of the sub-graphs, based on the each of the sub-graphs; and
step 9, repeating step 7 and step 8 through multiple iterations, where each of the multiple iterations adopts the exchange distributed mode in the step 7 to iteratively solve the optimization model in the step 8; and in a situation that each of the sub-graphs being meeting an iterative convergence condition, the multiple iterations are stopped, and a corresponding sub-graph of the sub-graphs meeting the iterative convergence condition is no longer updated while other sub-graphs of the sub-graphs not meeting the iterative convergence condition continue to participate the multiple iterations.
In a further embodiment, in the step 4, the constructing the skeleton graph based on the multiple superpixel regions, includes:
modeling the multiple superpixel regions as multiple skeleton graph nodes respectively, where a graph signal of each skeleton graph node of the multiple skeleton graph nodes is an average value of values of all pixels in a superpixel region of the multiple superpixel regions corresponding to the skeleton graph node, a coordinate of the skeleton graph node is a coordinate of a center pixel of the all pixels whose intensity is closest to the graph signal; and
constructing the skeleton graph :=({circumflex over (V)}s,
,
) based on the multiple skeleton graph nodes using a modified k-Nearest Neighbor (KNN) algorithm, where {circumflex over (V)}s represents a node set composed of the multiple skeleton graph nodes,
represents an edge set composed of edges of the skeleton graph, and
represents a skeleton graph adjacency matrix, which is used to describe a correlation
weight matrix of the edges of the skeleton graph,
where the skeleton graph adjacency matrix is determined by:
determining an initial skeleton graph adjacency matrix based on an element ai,j of the initial skeleton graph adjacency matrix, the element ai,j is expressed as a formula (1):
where Ii represents a coordinate of the i-th superpixel of all pixels of each of the multiple superpixel regions; Ij represents a coordinate of the j-th superpixel of all pixels of each of the multiple superpixel regions; xi represents a graph signal of the i-th superpixel of all pixels of each of the multiple superpixel regions; xj represents a graph signal of the j-th superpixel of all pixels of each of the multiple superpixel regions; and σ1 and σx represent two constant parameters; and
setting first k relative greater values of each row of the initial skeleton graph adjacency matrix to 1, setting the rest of the row of the initial skeleton graph adjacency matrix to 0, and symmetrizing and modifying the initial skeleton graph adjacency matrix after setting into an unweighted matrix as the skeleton graph adjacency matrix .
In a further embodiment, in the step 5, the m-th superpixel region of the multiple superpixel regions has Km pixels, and the constructing the local graph for all pixels of each of the multiple superpixel regions, to obtain the multiple local graphs respectively corresponding to the multiple superpixel regions, includes:
modeling all pixels of each of the multiple superpixel regions as pixel nodes, to thereby construct the multiple local graphs :=(
,
,
),
where represents a node set composed of all pixel nodes in the m-th superpixel region,
represents an edge set composed of edges for connecting each pixel node of the all pixel nodes in the m-th superpixel region to surrounding four neighbors of the pixel node, and
represents an unweighted four-neighborhood adjacency matrix; where a graph signal of each of the multiple local graphs is a vector xm=[xm,1,xm,2, . . . ,xm,K
K
K
where a matrix is formed based on graph signals of all bands, as a graph signal of the hyperspectral image, expressed as: Xm=[Xm(i,b)]=[Xm,1, . . . , Xm,b, . . . , Xm,p)]∈K
where Xm(i,b) represents the i-th graph signal of the image corresponding to the b-th band of the hyperspectral image; Xm,b represents the m-th graph signal of the image corresponding to the b-th band of the hyperspectral image, and K
In a further embodiment, in the step 6, a hyperspectral restoration problem is transformed into solving of multiple small-scale problems, each sub-graph of the multiple small-scale problems is located on one of the sub-graphs, and each sub-graph is centered on one of the multiple skeleton graph nodes of the skeleton graph;
where the f-th sub-graph of the sub-graphs is expressed as a formula (2):
=(
where
represents the f-th sub-graph,
where (f, d):={j∈{circumflex over (V)}s:ρ(f, j)≤d} represents a set of neighbor sub-graph nodes of a sub-graph node f of the f-th sub-graph within d hops;
where ρ(f,j) represents a shortest path from the sub-graph node f to a sub-graph node j;
where each sub-graph includes at least two of the multiple local graphs, the m-th local graph of the at least two of the multiple local graphs is one sub-graph node of the set of neighbor sub-graph nodes of the sub-graph node f corresponding to the f-th sub-graph within the d hops, that is m∈(f, d);
where d represents the set of neighbor sub-graph nodes within the d hops, m represents the m-th local graph and is a sub-graph node of the f-th sub-graph, in represents the node set composed of all pixel nodes in the m-th superpixel region of the multiple superpixel regions,
(f, d) represents the set of neighbor sub-graph nodes of the sub-graph node f corresponding to the f-th sub-graph of the skeleton graph
within the d hops;
where the edge set
nodes of the boundary node, and forming an edge between the pixel nodes of a local graph and the local graph
, if there is a boundary connection for the local graph
and the local graph
in corresponding superpixel regions thereof; or, forming only the edge between the pixel nodes of the local graph
and the local graph
, if there is no boundary connection for the local graph
and the local graph
in corresponding superpixel regions thereof, that is, there is edge connection between the pixel nodes of the local graph
and the local graph
;
where a Laplacian matrix is obtained, and a same sub-graph is applied to all bands of the hyperspectral image corresponding to the sub-graph;
where a graph signal of the sub-graph node ν∈ΣK
where m∈(f,d),b=1, . . . , p, Xf(i,b) represents a graph signal of the i-th sub-graph node corresponding to the b-th band in the f-th sub-graph
, Xf(1,b) represents a graph signal of a first sub-graph node corresponding to the b-th band in the f-th sub-graph
, Km represents a total number of pixels in the superpixel region corresponding to the m-th local graph on the f-th sub-graph
, and b represents the b-th band; and
where a matrix is formed based on graph signals of all bands using a formula expressed as follows: Xf=[Xf(i,b)]=[Xf,1, . . . , Xf,b, . . . , Xf,p)]∈(ΣK
(f, d), and thus the graph signal of the hyperspectral image on the sub-graph
, f=1, 2, . . . , Ms is obtained.
In a further embodiment, in the step 7, multiple computing units are set, each of the plurality of computing units corresponds to one of the multiple local graphs, and a transmission link with neighbored computing units of the computing unit is established by the computing unit based on edges of the skeleton graph, and
where the applying the exchange distributed mode to each of the sub-graphs to perform distributed denoising on the hyperspectral image, includes:
receiving computing results from the neighbored computing units when each of multiple iterations is performed;
fusing and averaging, by the computing unit, the received computing results from the neighbored computing units with a last calculation result of the computing unit;
determining, by the computing unit, an optimized computing result of the computing unit and optimized computing results of the neighbored computing units based on the optimization model of each of the sub-graphs; and
transmitting, by the computing unit, the optimized computing result of the computing unit and a corresponding one of the optimized computing results of the neighbored computing units to a corresponding one of the neighbored computing units. In a further embodiment, in the step 8, the optimization model is expressed as a formula (3):
where m∈(f, d), Km represents a total number of pixels in the m-th superpixel region, α and β are two parameters, and used to adjust proportions of a nuclear norm, GLR and an L1 norm, Sf represents a model of sparse noise of in the f-th sub-graph, Of represents hyperspectral image data originally input in the f-th sub-graph, ϵ represents a threshold value, r is a constant for indicating constraint on a rank of a hyperspectral signal; f represents an integer;
where GLR of signals of one band of the HSI is expressed as a formula (4):
∥x∥GLR=xTx=Σi˜jαi,j·(xi−xj)2 (4),
the GLR of signals of all bands of the HSI is expressed as a formula (5):
∥X∥GLR=Tr(XTX)=∥x1∥GLR+ . . . +∥xb∥GLR+ . . . +∥xp∥GLR (5),
where represents a Laplacian matrix of the f-th sub-graph, αi,j represents the (i, j)-th element of an adjacency matrix of the f-th sub-graph; xi represents the i-th graph signal of the f-th sub-graph; ∥xb∥GLR represents a GLR corresponding to the b-th band; and Tr(·) represents a trace of a matrix.
In a further embodiment, in the step 8, the optimization model is obtained according to augmented Lagrange method;
where the formula (3) is equivalently converted into a formula (6):
where Hf represents an introduced auxiliary variable, and Hf=Xf;
where an augmented Lagrange function of the formula (6) is expressed as a formula (7):
where r represents a penalty variable, which is greater than zero, and Cf and Df represent Lagrange multipliers;
where the formula (7) is alternately optimized through following updatings:
1) updating Hf
where (k+1) represents the (k+1)-th iteration, Xf(k) represents the denoised hyperspectral image after the k-th iteration of the f-th sub-graph, Sf(k) represents sparse noise after the k-th iteration of the f-th sub-graph, Cf(k) and Df(k) represent Lagrange multipliers after the k-th iteration of the f-th sub-graph;
where for a given matrix Z ∈ K×p with a rank r, singular value decomposition is used to decompose the given matrix, to obtain a formula (9):
Z=UE
r
V
H
,E
r=diag(σi) (9),
where U ∈ K×r and V ∈
p×r are orthogonal matrices, Er represents a diagonal matrix composed of singular values, a singular value σi(1≤i≤r)≥0, a threshold value ϕ being greater than or equal to zero is given, and a singular value contraction operator is defined as a formula (10):
D
ϕ(Z)=UDϕ(Er)VH (10)
D
ϕ(Er)=diag(max(η, 0))
where η=σi−ϕ, η represents a vector composed of a difference between the i-th singular value σi and a threshold value ϕ, σi represents the i-th singular value after singular value decomposition, ϕ represents a constant, which is the threshold value, Dϕ(·) represents a solution of a singular value contraction operator, the singular value contraction operator (10) is a solution of a problem (11):
where the singular value contraction operator (10) is used to optimize a solution of the formula (8) to obtain a formula (12):
2) updating Xf
where Bf=Hf(k+1)−DB(k)/γ+[b1, . . . , bp], according to the matrix form of GLR in the formula (5), the formula (13) is equivalent to a formula (14):
where the formula (14) is decomposed into p sub-problems and solved band by band, and the j-th sub-problem of the p sub-problems is expressed as a formula (15):
where the formula (15) is a least squares problem, and a solution of the formula (15) is
expressed as a formula (16):
x
j
(k+1)=(2α+γI)−1γbj (16),
where a solution of the formula (14) is obtained by combining the solutions (16) of each band and is expressed as a formula (17):
where the formula (17) is rewritten in matrix form expressed in a formula (18):
X
f
(k+1)=(2α+γI)−1Bf (18),
where by derivation, the formula (13) is equivalently transformed to a formula (19):
3) updating Sf
where a soft threshold operator is introduced and expressed as a formula (21):
where φ ∈, Δ>0, and a solution of the formula (20) is expressed as a formula (22):
4) updating the Lagrange multipliers Cf and Df, and the penalty variable γ
C
f
(k+1)
=C
f
(k)+γ(Of−Hf(k+1)−Sf(k+1))
D
f
(k+1)
=D
f
(k)+γ(Xf(k+1)−Hf(k+1)) (23 )
γ(k+1)=min(ργ(k)+γmax),
where ρ represents a growth step, and γmax represents a constant.
In a further embodiment, in the step 9, the iterative convergence condition is determined according to a formula (24):
∥Of−Hf(k+1)−Sf(k+1)∥F/∥Of∥F≤ϵ1 (24)
∥Hf(k+1)−Xf(k+1∥∞≤ϵ2,
where ϵ1 and ϵ2 represent two constants, Hf(k+1) represents introduced auxiliary variable and is a solution of Hf=Xf after (k+1) iterations, Sf(k+1) represents a solution of sparse noise after (k+1) iterations of the f-th sub-graph, and Xf(k+1) represents a solution of a clean hyperspectral image after (k+1) iterations of the f-th sub-graph.
A hyperspectral image distributed restoration system based on graph signal processing and superpixel segmentation is provided according to an embodiment of the present disclosure, including: a memory, a processor, and a computer program stored in the memory and executable by the processor, where the hyperspectral image distributed restoration method based on graph signal processing and superpixel segmentation described above is implemented when the computer program is executed by the processor.
The present disclosure may have at least one of the following beneficial effects.
According to the large-scale hyperspectral image distributed restoration method based on graph signal processing and superpixel segmentation provided by the present disclosure, firstly, the underlying structure of the HSI is modeled by a two-layer architecture graph. The upper layer, called skeleton graph, is a rough graph constructed by using the modified k-nearest-neighborhood algorithm and its nodes correspond to a series of superpixels formed by HSI segmentation. The skeleton graph can efficiently characterize the inter-correlations between superpixels, while preserving the boundary information and reducing the computational complexity. The lower layer, called detailed graph consisting of a series of local graphs which are constructed to model the similarities between pixels. Second, based on the two-layer graph architecture, the HSI restoration problem is formulated as a series of optimization problems each of which resides on a subgraph. In each optimization problem, a graph Laplacian regularization is defined and incorporated into a low-rank-based model. Third, a novel distributed algorithm is tailored for the restoration problem, by using the information interaction between the nodes of skeleton graph and subgraphs. The present method overcomes the problem that hyperspectral images (HSI) are often disturbed by various noises and thus brings great challenges to subsequent applications.
A distributed algorithm based on graph signal processing may serve as a new idea to solve this problem. In the graph signal processing framework, these distributed methods always exchange information with neighbors during each iteration, with a potential to achieve better denoising performance than methods based on non-local or superpixel segmentation. Numerical experiments conducted on both synthetic and real world datasets demonstrate the effectiveness of the proposed restoration algorithm compared with existing methods.
Other advantages, objectives, and features of the present disclosure will be set forth in the following specification to some extent, and will be apparent to those skilled in the art based on the following investigation to some extent, or may be taught from the practice of the present disclosure. The objects and other advantages of the present disclosure can be realized and obtained by the following description.
In order to make the objectives, the technical solutions and the beneficial effects of the present disclosure clearer, the present disclosure provides the following drawings for illustration.
The present disclosure is further described hereinafter combined with accompanying drawings and specific embodiments, so that those skilled in the art can better understand the present disclosure and implement the present disclosure, but the specific embodiments hereinafter are not intended to limit the present disclosure.
The first embodiment of the present disclosure provides a large-scale HSI distributed restoration method based on graph signal processing and superpixel segmentation. The method may include steps 1 through step 9 as follows.
In step 1, an input signal model is constructed, to linearly normalize an HSI to a range from 0 to 1, to obtain a normalized HSI. The constructed input signal model Y=X+S+N, where Y presents an input noise signal; X represents a clean original image; S represents sparse noise, which may be an impulse noise, a bad dot, a stripe noise; and N represents a Gaussian noise.
In step 2, an image corresponding to a target-band (for example, a specified band or a random band) from the normalized HSI is selected, and the image is pre-denoised using a median filter to obtain a pre-denoised image.
In step 3, the pre-denoised image is segmented into multiple superpixel regions using simple linear iterative cluster (SLIC) superpixel segmentation algorithm.
In step 4, a skeleton graph is constructed based on the multiple superpixel regions. It is assumed that there are Ms superpixel regions after the segmenting. Ms superpixel regions are modeled as Ms skeleton graph nodes respectively. A graph signal of each skeleton graph node is an average value of all pixels in a superpixel region of the multiple superpixel regions corresponding to the skeleton graph node, a coordinate of the skeleton graph node is a coordinate of a target pixel in the superpixel region, and an intensity value of the target pixel is closest to the graph signal. The skeleton graph :=({circumflex over (V)}s,
,
) is constructed based on the multiple skeleton graph nodes using a modified k-Nearest Neighbor (KNN) algorithm, where {circumflex over (V)}s represents a node set composed of the multiple skeleton graph nodes,
represents an edge set composed of edges of the skeleton graph, and
represents a skeleton graph adjacency matrix, which is used to describe a correlation weight matrix of the edges of the skeleton graph, and the skeleton graph adjacency matrix
is determined by:
determining an initial skeleton graph adjacency matrix based on an element ai,j of the initial skeleton graph adjacency matrix, the element ai,j is expressed as a formula (1):
where Ii represents a coordinate of the i-th superpixel of all pixels of each of the multiple superpixel regions; Ij represents a coordinate of the j-th superpixel of all pixels of each of the multiple superpixel regions; xi represents a graph signal of the i-th superpixel of all pixels of each of the multiple superpixel regions; xj represents a graph signal of the j-th superpixel of all pixels of each of the multiple superpixel regions; and σ1 and σx represent two constant parameters; and
setting first k relative greater values of each row of the initial skeleton graph adjacency matrix to 1, setting the rest of the row of the initial skeleton graph adjacency matrix to 0, and symmetrizing and modifying the initial skeleton graph adjacency matrix after setting into an unweighted matrix as the skeleton graph adjacency matrix .
In step 5, a local graph is constructed for all pixels of each of the multiple superpixel regions, to obtain multiple local graphs respectively corresponding to the multiple superpixel regions. The m-th superpixel region of the multiple superpixel regions has Km pixels. All pixels of each of the multiple superpixel regions are modeled as pixel nodes, to thereby construct the multiple local graphs :=(
,
,
).
represents a node set composed of all pixel nodes in the m-th superpixel region,
represents an edge set composed of edges for connecting each pixel node of the all pixel nodes in the m-th superpixel region to surrounding four neighbors of the pixel node, and
represents an unweighted four-neighborhood adjacency matrix, which is a sparse matrix. A graph signal of each of the multiple local graphs is a vector xm=[xm,1,xm,2, . . . ,xm,K
K
K
Xm(i,b) represents the i-th graph signal of the image corresponding to the b-th band of the hyperspectral image; Xm,b represents the m-th graph signal of the image corresponding to the b-th band of the hyperspectral image, and K
In step 6, sub-graphs are constructed based on the skeleton graph and the multiple local graphs. Since each sub-graph corresponds to one computing node, a hyperspectral restoration problem is transformed into solving of multiple small-scale problems, each sub-graph of the multiple small-scale problems is located on one of the sub-graphs, and each sub-graph is centered on one of the multiple skeleton graph nodes of the skeleton graph. The f-th sub-graph of the sub-graphs is expressed as a formula (2):
:=(
represents the f-th sub-graph,
composed of all sub-graph nodes of the f-th sub-graph,
(f, d):={j∈{circumflex over (V)}s:ρ(f, j)≤d} represents a set of neighbor sub-graph nodes of a sub-graph node f of the f-th sub-graph within d hops; ρ(f, j) represents a shortest path (i.e., a quantity of the hopes) from the sub-graph node f to a sub-graph node j. Each sub-graph includes multiple local graphs, the m-th local graph of the at least two of the multiple local graphs is one sub-graph node of the set of neighbor sub-graph nodes of the sub-graph node f corresponding to the f-th sub-graph within the d hops, that is m∈
(f, d); where d represents the set of neighbor sub-graph nodes within the d hops, m represents the m-th local graph and is a sub-graph node of the f-th sub-graph,
represents the node set composed of all pixel nodes in the m-th superpixel region of the multiple superpixel regions,
(f,d) represents the set of neighbor sub-graph nodes of the sub-graph node f corresponding to the f-th sub-graph of the skeleton graph
within the d hops.
The edge set and the local graph
, if there is a boundary connection for the local graph
and the local graph
in corresponding superpixel regions thereof; or, forming only the edge between the pixel nodes of the local graph
and the local graph
, if there is no boundary connection for the local graph
and the local graph
in corresponding superpixel regions thereof, that is, there is edge connection between the pixel nodes of the local graph
and the local graph
.
A Laplacian matrix is obtained, and a same sub-graph is applied to all bands of the hyperspectral image corresponding to the sub-graph.
A graph signal of the sub-graph node ν ∈ ΣK
(f,d),b=1, . . . , p, Xf(i,b) represents a graph signal of the i-th sub-graph node corresponding to the b-th band in the f-th sub-graph
, Xf(1,b) represents a graph signal of a first sub-graph node corresponding to the b-th band in the f-th sub-graph
, Km represents a total number of pixels in the superpixel region corresponding to the m-th local graph on the f-th sub-graph
, and b represents the b-th band.
A matrix is formed based on graph signals of all bands using a formula expressed as follows: Xf=[Xf(i,b)]=[Xf,1, . . . , Xf,b, . . . , Xf,p)]∈(ΣK
(f,d), and thus the graph signal of the hyperspectral image on the sub-graph
, f=1, 2, . . . , Ms is obtained.
In step 7, a receiving-transmitting-fusing (also referred to as exchange) distributed mode is applied to each of the sub-graphs, to perform distributed denoising on the HSI. It is assumed that there are Ms computing units, each computing unit corresponds to one of the multiple local graphs , and a transmission link with neighbored computing units of the computing unit is established by the computing unit based on edges of the skeleton graph
.
Ms computing units perform following steps during each iteration of the optimization model of each of the sub-graphs: receiving computing results from the neighbored computing units when each of multiple iterations is performed; fusing and averaging, by the computing unit, the received computing results from the neighbored computing units with a last calculation result of the computing unit; determining, by the computing unit, an optimized computing result of the computing unit and optimized computing results of the neighbored computing units based on the optimization model of each of the sub-graphs; and transmitting, by the computing unit, the optimized computing result of the computing unit and a corresponding one of the optimized computing results of the neighbored computing units to a corresponding one of the neighbored computing units. The above steps are repeated for iteration solving of the optimization model of each of the sub-graphs.
In step 8, an optimization model of each of the sub-graphs is established based on the each of the sub-graphs; the optimization model is expressed as a formula (3):
m∈(f, d), Km represents a total number of pixels in the m-th superpixel region, α and β are two parameters, and used to adjust proportions of a kernel norm, GLR and an L1 norm, Sf represents a model of sparse noise of in the f-th sub-graph, Of represents hyperspectral image data originally input in the f-th sub-graph, ϵ represents a threshold value, r is a constant for indicating constraint on a rank of a hyperspectral signal; f represents an integer.
GLR of signals of one band of the HSI is expressed as a formula (4):
∥x∥GLR=xTx=Σi˜jαi,j·(xi−xj)2 (4).
The GLR of signals of all bands of the HSI is expressed as a formula (5):
∥X∥GLR=Tr(XTX)=∥x1∥GLR+ . . . +∥xb∥GLR+ . . . +∥xp∥GLR (5).
Therefore, it can be regarded as the combination vector form of GLR. GLR is a quadratic form of matrix, which has many excellent characteristics. A solution of quadratic problem can be solved by simple standard linear algebra. In addition, because of the rank constraint in the optimization model, the formula (3) is non-convex, and the augmented Lagrange multiplier method (ALM) can be used to solve this problem.
represents a Laplacian matrix of the f-th sub-graph, αi,j represents the (i, j)-th element of an adjacency matrix of the f-th sub-graph; xi represents the i-th graph signal of the f-th sub-graph; ∥xb∥GLR represents a GLR corresponding to the b-th band; and Tr(·) represents a trace of a matrix.
The formula (3) is equivalently converted into a formula (6):
where Hf represents an introduced auxiliary variable, and Hf=Xf.
An augmented Lagrange function of the formula (6) is expressed as a formula (7):
where r represents a penalty variable, which is greater than zero, and Cf and Df represent Lagrange multipliers. When the formula (7) is alternately optimized, one variable can be optimized while other variables are remain unchanged.
The formula (7) is alternately optimized through following updatings:
1) updating Hf
(k+1) represents a (k+1)-th iteration, Xf(k) represents the denoised hyperspectral image after the k-th iteration of the f-th sub-graph, Sf(k) represents sparse noise after the k-th iteration of the f-th sub-graph, Cf(k) and Df(k) represent Lagrange multipliers after the k-th iteration of the f-th sub-graph.
For a given matrix Z ∈ K×p with a rank r, singular value decomposition is used to decompose the given matrix, to obtain a formula (9):
Z=UE
r
V
H
,E
r=diag(σi) (9),
U ∈ K×r and V ∈
p×r are orthogonal matrices, Er represents a diagonal matrix composed of singular values, a singular value σi(1≤i≤r)≥0, a threshold value ϕ being greater than or equal to zero is given, and a singular value threshold value (also referred to as ingular value contraction) operator is defined as a formula (10):
D
ϕ(Z)=UDϕ(Er)VH (10)
D
ϕ(Er)=diag(max(η,0))
η=σi−ϕ, η represents a vector composed of a difference between an i-th singular value σi and a threshold value ϕ, σi represents the i-th singular value after singular value decomposition, ϕ represents a constant, which is the threshold value, Dϕ(·) represents a solution of a singular value contraction operator, the singular value contraction operator (10) is a solution of a problem (11):
The singular value contraction operator (10) is used to optimize a solution of the formula (8) to obtain a formula (12):
2) updating Xf
For convenience, Bf=Hf(k+1)−DB(k)/γ+[b1, . . . , bp], and according to the matrix form of GLR in the formula (5), the formula (13) is equivalent to a formula (14):
Then, the formula (14) is decomposed into p sub-problems and solved band by band, and the j-th sub-problem of the p sub-problems is expressed as a formula (15):
The formula (15) is a least squares problem, and a solution of the formula (15) is expressed as a formula (16):
x
j
(k+1)=(2α+γI)−1γbj (16).
Therefore, a solution of the formula (14) can be obtained by combining the solutions (16) of each band and is expressed as a formula (17):
It is apparent that the formula (17) can be rewritten in matrix form expressed in a formula (18):
X
f
(k+1)=(2α+γI)−1Bf (18)
By derivation, the formula (13) is equivalently transformed to a formula (19):
3) updating Sf
A soft threshold operator is introduced and expressed as a formula (21):
where φ∈, Δ>0, and a solution of the formula (20) is expressed as a formula (22):
4) updating the Lagrange multipliers Cf and Df, and the penalty variable γ
C
f
(k+1)
=C
f
(k)+γ(Of−Hf(k+1)−Sf(k+1))
D
f
(k+1)
=D
f
(k)+γ(Xf(k+1)−Hf(k+1)) (23)
γ(k+1)=min(ργ(k)+γmax),
where ρ represents a growth step, and γmax represents a constant.
In step 9, the step 7 and the step 8 are repeated through multiple iterations, where each of the multiple iterations adopts the receiving-transmitting-fusing distributed mode in the step 7 to iteratively solve the optimization model in the step 8; and in a situation that each of the sub-graphs being meeting an iterative convergence condition, the repeating is stopped, and a corresponding sub-graph of the sub-graphs meeting the iterative convergence condition is no longer updated while other sub-graphs of the sub-graphs not meeting the iterative convergence condition continue to participate the repeating. The iterative convergence condition is determined according to a formula (24):
∥Of−Hf(k+1)−Sf(k+1)∥F/∥Of∥F≤ϵ1 (2)
∥Hf(k+1)−Xf(k+1)∥∞≤ϵ2,
where ϵ1 and ϵ2 represent two constants, Hf(k+1) represents introduced auxiliary variable and is a solution of Hf=Xf after (k+1) iterations, Sf(k+1) represents a solution of sparse noise after (k+1) iterations of the f-th sub-graph, and Xf(k+1) represents a solution of a clean hyperspectral image after (k+1) iterations of the f-th sub-graph.
Further, a hyperspectral image distributed restoration system based on graph signal processing and superpixel segmentation is provided according to an embodiment of the present disclosure, including: a memory, a processor, and a computer program stored in the memory and executable by the processor, where the hyperspectral image distributed restoration method based on graph signal processing and superpixel segmentation described above is implemented when the computer program is executed by the processor.
In the second embodiment, the input hyperspectral image is a clean hyperspectral image of a Washington Shopping Center, namely X, which has 191 bands and a dimension thereof is 1280×307×191. Noises are added artificially, and the added noises include parse noise (including a salt and pepper noise and a deadline noise) and a Gaussian noise. An output after noise addition is Y. A three-dimensional (3D) hyperspectral image after noise addition is input into the model of this method of the present disclosure to obtain the denoised hyperspectral image.
In this embodiment, parameters of the specific steps are set as follows. In the step (2), a specific hyperspectral band is selected, and hyperspectral data of the 50-th band is taken; and a window adopted by the median filtering is [3, 3]. For parameters of the SLIC superpixel segmentation in the step (3), a size of the regions is set to 30, and a regularization parameter is set to 0.05. In the step (4), σ1=100, σx=0.2, and k=3. In the step (6), d=1. In the step (8), α=0.5, β=0.23, ϵ=ϵ1 =ϵ2=1e−6, r=3, ρ=1.5, γmax=1e6, each of Hf(0), Xf(0), Sf(0), Cf(0), Df(0) is an all-zero matrix, and γ(0)=1e−2. Finally, a peak signal-to-noise ratio (PSNR) and a structural similarity (SSIM) parameters of the denoised hyperspectral image X and the clean image without adding noise are calculated to evaluate the superiority of the denoised effect.
The denoised 3D hyperspectral image is input into the technical solution model with set parameters of the present disclosure. The solution GLRSSDA is compared with a hyperspectral image restoration method based on low-rank matrix recovery (LRMR) proposed by Hongyan Zhang, the hyperspectral image restoration method based on total-variation-regularized low-rank matrix factorization (LRTV) proposed by Wei He and the hyperspectral image restoration method based on superpixel segmentation of smooth band (SSSB) proposed by Yaru Fan, and the comparison graphs before and after denoising are shown in
In summary, it can be observed that the technical solution of the present disclosure is superior to the technical solutions of the above-mentioned prior arts. The PSNR and SSIM of each band in the technical solution of the present disclosure and the above-mentioned existing solution are averaged to obtain MPSNR and MSSIM indexes, to obtain results shown in a table 1 below, and it can be observed that the technical solution of the present disclosure is superior to the above-mentioned existing solutions.
The large-scale hyperspectral image distributed restoration method based on graph signal processing and superpixel segmentation provided by the embodiment of the present disclosure overcomes the problem that hyperspectral images (HSI) are often disturbed by various noises.
The above-mentioned embodiments are merely preferred embodiments for fully explaining the present disclosure, and the scope of protection of the present disclosure is not limited thereto. Equivalent substitutions or changes made by those skilled in the art on the basis of the present disclosure are within the scope of protection of the present disclosure. The scope of protection of the disclosure is subject to the claims.
Number | Date | Country | Kind |
---|---|---|---|
2022109338569 | Aug 2022 | CN | national |