Disclosure of Invention
The invention aims to provide an SAR target recognition method based on a Bayesian multi-core learning support vector machine aiming at the defects of the prior art so as to improve the target recognition performance.
The technical scheme of the invention is realized as follows:
technical thought
The invention combines Bayesian inference with a multi-core learning method, introduces the multi-core learning method aiming at the selection problem of different data characteristics, and has good generalization capability and stronger learning capability; meanwhile, Bayesian inference is used for reasoning out the solution of the original problem of the support vector machine. The implementation scheme is as follows: firstly, preprocessing an original SAR image to obtain three data characteristics of an image domain, a frequency domain and a sparse coefficient, and respectively calculating a kernel matrix corresponding to a radial kernel function RBF; thirdly, combining the three kernel matrixes by using a multi-kernel learning algorithm; finally, a Bayesian model is inferred by using the combined kernel matrix of the training data to obtain an optimal solution, and the combined kernel matrix of the test data is classified, wherein the method comprises the following implementation steps:
(A) SAR image preprocessing step:
A1) inputting an original SAR image: i ═ ImnM is more than or equal to 1 and less than or equal to M, N is more than or equal to 1 and less than or equal to N, wherein imnRepresenting the amplitude pixel value of an original SAR image, wherein M represents the row number of the SAR image, and N represents the column number of the SAR image;
A2) performing binary segmentation on the original SAR image I, and calculating to obtain the centroid of the SAR image
A3) Circularly shifting the original SAR image I to make the center of massMoving to the central position of the image to obtain a registration image I1;
A4) To the registration image I1Sequentially carrying out logarithmic transformation, median filtering and image interception to obtain image domain characteristics I of the SAR image2And image domain feature I2Carrying out column vectorization;
A5) to the registration image I1Image interception and two-dimensional Fourier transform are carried out, and zero frequency is shifted to the center of the image to obtain frequency domain characteristics I3And the frequency domain characteristic I3Carrying out column vectorization;
A6) respectively repeating the processes A1) -A4) on the SAR image training set and the test set to obtain a training data set T of the image domain characteristicstrAnd test data set Tte;
A7) Respectively repeating the processes A1) -A5) on the SAR image training set and the test set to obtain a training data set P of frequency domain characteristicstrAnd a test data set Pte;
A8) Training set T for image domain features using KSVD algorithmtrLearning to obtain dictionaries D and TtrCorresponding sparse coefficient feature training data set StrTesting the data set T in conjunction with the dictionary D and the image domain featuresteUsing OMP algorithm to calculate and obtain sparse coefficient characteristic test data set Ste;
(B) A multi-core learning step:
B1) training data set T using radial kernel function RBF in combination with image domain featurestrAnd test data set TteAnd calculating to obtain a kernel matrix K of the image domain characteristic training data setttr(Ttr,Ttr) Sum image domain feature test dataset kernel matrix Ktte(Ttr,Tte);
B2) Training a data set P using radial kernel function RBF in combination with frequency domain featurestrAnd a test data set PteAnd calculating to obtain a kernel matrix K of the frequency domain characteristic training data setptr(Ptr,Ptr) Sum frequency domain feature test dataset kernel matrix Kpte(Ptr,Pte);
B3) Training data set S by using radial kernel function RBF and combining sparse coefficient featurestrAnd a test data set SteAnd calculating to obtain a kernel matrix K of the sparse coefficient characteristic training data setstr(Str,Str) And sparse coefficient feature test data set kernel matrix Kste(Str,Ste);
B4) Combining the training set kernel matrix and the test set kernel matrix of the three characteristics obtained by calculation in the steps B1) -B3), and calculating by using a kernel combination method to obtain a combined kernel matrix K of the SAR image training settr(V',Vtr) And combined kernel matrix K of test sette(V',Vte) Wherein V' represents a set of basis vectors, VtrRepresenting a SAR image training data set, VteRepresenting a SAR image test dataset;
(C) bayesian inference steps:
C1) combined kernel matrix K using SAR image training settr(V',Vtr) Establishing a Bayesian multi-core learning support vector machine model;
C2) solving the Bayes multi-core learning support vector machine model by using an expectation maximization algorithm EM to obtain an optimal solution β' of the Bayes multi-core learning support vector machine model;
C3) using the optimal solution β' of the Bayesian multi-kernel learning support vector machine model obtained in the step C2), and combining the combined kernel matrix K of the SAR image test sette(V',Vte) And calculating to obtain the target class label y of the SAR imagete。
Compared with the prior art, the invention has the following advantages:
the invention combines a Bayesian support vector machine model with a multi-core learning method, and provides the SAR target recognition method based on the Bayesian multi-core learning support vector machine, so that the multi-core learning method is superior to a single-core learning method in the aspect of selecting data characteristics, the relevance among different data characteristics can be embodied, and the target recognition performance is obviously improved.
Detailed Description
The following steps and effects of the present invention will be further explained with reference to the accompanying drawings:
the implementation steps of the invention are as follows with reference to fig. 1.
Step 1, preprocessing the SAR image and calculating a kernel matrix.
1a) Inputting a raw SAR image as shown in fig. 2 (a): i ═ ImnM is more than or equal to 1 and less than or equal to M, N is more than or equal to 1 and less than or equal to N, wherein imnRepresenting the amplitude pixel value of an original SAR image, wherein M represents the row number of the SAR image, and N represents the column number of the SAR image;
1b) performing binary segmentation on the original SAR image I by using a variable power Ostu segmentation algorithm to obtain a segmented SAR image I';
1c) performing point multiplication calculation on the segmented SAR image I' and the original SAR image I to obtain a point-multiplied SAR image as shown in FIG. 2(b), and calculating the centroid of the point-multiplied SAR image
In formula (II) to'mnA pixel value representing the point-multiplied SAR image;
1d) circularly shifting the original SAR image I to make the center of massMoving to the central position of the image to obtain a registered SAR image I1As shown in FIG. 2 (c);
1e) to the registration SAR image I1Carrying out image interception to obtain an intercepted SAR image I0As shown in FIG. 2 (d);
1f) for intercepting SAR image I0Performing logarithmic transformation to obtain a logarithmic SAR image I' "as shown in FIG. 2 (e);
1g) carrying out median filtering processing on the logarithmic SAR image I' ″ to obtain image domain characteristics I2As shown in FIG. 2(f), image domain feature I2Carrying out column vectorization;
1h) to the registration SAR image I1Image interception and two-dimensional Fourier transform are carried out, and zero frequency is shifted to the center of the image to obtain frequency domain characteristics I3And the frequency domain characteristic I3Carrying out column vectorization;
1i) respectively repeating 1a) -1 g) on the training set and the testing set of the original SAR image to obtain a training data set T of the image domain characteristicstrAnd test data set Tte;
1j) Respectively repeating 1a) -1 h) on the training set and the test set of the original SAR image to obtain a training data set P of the frequency domain characteristicstrAnd a test data set Pte;
1k) Training set T for image domain features using KSVD algorithmtrLearning to obtain dictionaries D and TtrCorresponding sparse coefficient feature training data set StrTesting the data set T in conjunction with the dictionary D and the image domain featuresteUsing OMP algorithm to calculate and obtain sparse coefficient characteristic test data set Ste。
And 2, performing multi-core learning calculation on the three SAR image characteristics obtained in the step 1.
2a) Training data set T using radial kernel function RBF in combination with image domain featurestrAnd test data set TteAnd calculating to obtain a kernel matrix K of the image domain characteristic training data setttr(Ttr,Ttr) Sum image domain feature test dataset kernel matrix Ktte(Ttr,Tte) Wherein the radial kernel function RBF is expressed as follows:
in the formula, q 'and q represent two data points in the same space, K (q', q) represents a calculated radial kernel function value, and sigma represents a radial kernel function parameter;
2b) training a data set P using radial kernel function RBF in combination with frequency domain featurestrAnd a test data set PteAnd calculating to obtain a kernel matrix K of the frequency domain characteristic training data setptr(Ptr,Ptr) Sum frequency domain feature test dataset kernel matrix Kpte(Ptr,Pte);
2c) Training data set S by using radial kernel function RBF and combining sparse coefficient featurestrAnd a test data set SteAnd calculating to obtain a kernel matrix K of the sparse coefficient characteristic training data setstr(Str,Str) And sparse coefficient feature test data set kernel matrix Kste(Str,Ste);
2d) Combining the training set kernel matrixes of the three characteristics obtained in the steps 2a) to 2c), and obtaining a combined kernel matrix K of the SAR image training set by using a kernel combination methodtr(V',Vtr):
Ktr(V',Vtr)=ηtKttr(Ttr,Ttr)+ηpKptr(Ptr,Ptr)+ηsKstr(Str,Str)
In the formula, ηtThe combination coefficient of the kernel matrix of the image domain characteristic data set is represented, the value is 0.5,
ηpthe combination coefficient of the nuclear matrix of the frequency domain characteristic data set is expressed, the value is 0.5,
ηsrepresenting the combined coefficients of a sparse coefficient feature kernel matrix; the value is 0.5;
2e) combining the test set kernel matrixes of the three characteristics obtained by calculation in the steps 2a) to 2c), and calculating by using a kernel combination method to obtain a combined kernel matrix K of the SAR image test sette(V',Vte):
Kte(V',Vte)=ηtKtte(Ttr,Tte)+ηpKpte(Ptr,Pte)+ηsKste(Str,Ste)。
And 3, constructing a Bayesian multi-core learning support vector machine model.
3a) Given sample setWherein x islRepresents a training sample, ylRepresenting a training index, q representing a sample dimension, L representing a sample number, and T representing a matrix transposition symbol; the unconstrained conditional expression of the maximum edge classifier of the support vector machine is as follows:
in the formula, a first term is a regular term, and a second term is a penalty term;the vector of the augmentation is represented by,representing the augmentation weight, and gamma representing a harmonic parameter;
3b) obtaining an augmented weight solution of a Support Vector Machine (SVM) maximum edge classifier according to Lagrange duality In the formula, αjRepresenting the lagrangian coefficient;
will be provided withSubstituting into the penalty term to obtain:introducing a mapping function phi (-) into the penalty term to obtain a penalty term expression after the mapping function is introduced:
ream type middle βj=αjyj,And obtaining a final punishment term expression:
in the formula,β=(β1,…,βj,…,βL),representing augmented vectorsAnd an augmented vectorKernel function value, Δ tableShowing a training sample matrix;
3c) constructing a regularization term ofAdding the regular term and the final penalty term to obtain a final objective function d (β) which is:
wherein κ represents a harmonic parameter;
3d) the exponent of the negative of the regularization term in the objective function is calculated and defined as the pseudo-prior distribution function:
3e) and calculating the index of the negative number of the final penalty term in the target function, and defining the index as a pseudo likelihood distribution function:
wherein y is (y)1,…,yl,…,yL) And Δ' represents a training sample matrix;
3f) obtaining a pseudo posterior distribution function according to the calculation results in the step 3d) and the step 3 e):
p(β|y,K(Δ,Δ'))∝p(β)p(y|β,K(Δ,Δ')),
3h) combined kernel matrix K for training data set by SAR imagetr(V',Vtr) Replacing K (delta, delta') in the step 3f), and establishing a Bayesian multi-core learning support vector machine model:
p(β|y,Ktr(V',Vtr))∝p(β)p(y|β,Ktr(V',Vtr))
in the formula, Ktr(V',Vtr)=(Ktr(V',v1),…,Ktr(V',vl),…,Ktr(V',vL)),Ktr(V',vl) Represents the vector formed by inner product calculation of the basis set and the training sample, p (y | β, K)tr(V',Vtr) Is a pseudo-likelihood distribution function after introducing a combined kernel matrix, wherein:
and 4, solving the Bayesian multi-core learning support vector machine model.
4a) By containing lambdalThe integral expression of (c) represents the pseudo-likelihood distribution function after introducing the combined kernel matrix in step 3 h):
wherein the pseudo-likelihood distribution function has the following relation:
p(yl|β,Ktr(V',vl))=∫p(yl,λl|β,Ktr(V',vl))dλl
wherein v islRepresents a training sample, λlRepresenting hidden variables, p (y)l,λl|β,Ktr(V',vl) Is a pseudo-likelihood distribution function after adding hidden variables;
4b) introducing a new variable lambda into the Bayesian multi-core learning support vector machine model according to the pseudo-likelihood distribution function after the hidden variable is added in the step 4a), and obtaining a new relational expression:
p(β,λ|y,Ktr(V',Vtr))∝p(β)p(y,λ|β,Ktr(V',Vtr))
wherein,
in the formula, λ represents an implicit variable vector, and λ ═ λ1,…,λl,…,λL);
4c) Obtaining a posterior distribution function of lambda according to the new relational expression in the step 4 b):
in the formula,
4d) obtaining lambda from the posterior distribution function of lambda obtained in 4c)lCondition posterior distribution of (1):
in the formula,represents a generalized inverse Gaussian distribution according toAndthe conversion relationship between the two is obtainedCondition a posteriori ofDistribution:
in the formula,representing an inverse gaussian distribution, representing obedient signs in the distribution function;
4e) according to 4d) inProperty of the conditional posterior distribution and inverse Gaussian distribution ofDesired value of (a):
4f) according to the new relational expression obtained in the step 4b) and the relational expression obtained in the step 4e)The expectation value is solved by using an expectation maximization algorithm EM to obtain an iterative solution β of the Bayesian multi-core learning support vector machine model(m+1)Expression (c):
wherein m represents the mth iteration number, I represents an identity matrix,to representThe mth expected iteration value of (1);
4g) setting the maximum iteration number as M ', repeating the step 4f), stopping iteration when the iteration number reaches M ', and finally obtaining the optimal solution β ' of the Bayesian multi-core learning support vector machine model:
and 5, calculating to obtain the target identification category label of the SAR image.
5a) Utilizing the optimal solution β' of the Bayesian multi-core learning support vector machine model obtained in the step 4g) in combination with the SAR image test data set Kte(V',Vte) Obtaining the SAR image target identification mark y by using the following formulate:
yte=sgn(β'TKte(V',Vte))
In the formula, sgn (·) represents a sign function.
At this point, classification of the SAR target is completed.
The effect of the present invention is further illustrated by the following experiment on measured data:
1. experimental scenarios and parameters:
the data used in the experiments were published dynamic and static target acquisition and identification MSTAR datasets. In the data set, model image data of BMP2SN9563, BTR70C71, and T72SN132 at a pitch angle of 17 ° are selected as training data, model image data of 7 at a pitch angle of 15 ° are selected as test data, and referred to as BMP2SN9566 and BMP2SNC21 as variants of BMP2SN9563, T72SNs7 and T72SN812 as variants of T72SN132, and the original image size is 128 × 128.
The data types and sample numbers used in this experiment are shown in table 1:
TABLE 1 MSTAR Experimental data
The experimental parameters were set as follows:
the size of the preprocessed SAR image is 63 multiplied by 63; radial kernel parameter sigma corresponding to image domain featuretFrequency domain feature corresponds to a radial kernel parameter σ as 1p0.1, radial kernel parameter σ corresponding to sparse coefficient features1, a new harmonic parameter k in the Bayesian multi-core learning support vector machine model is 0.01;
2. the content and the result of the experiment are as follows:
the invention and other existing 5 methods classify the MSTAR three kinds of data sets, wherein the 1 st kind is a linear support vector machine, the 2 nd kind is a single-core learning support vector machine, the 3 rd kind is a multi-core learning support vector machine, the 4 th kind is a Bayes support vector machine, and the 5 th kind is a Bayes single-core learning support vector machine;
the experimental steps for target identification by the method of the invention are as follows:
firstly, preprocessing three types of MSTAR data in an experiment, and calculating by using a radial kernel function to obtain three characteristic kernel matrixes, namely an image domain characteristic kernel matrix, a frequency domain characteristic kernel matrix and a sparse coefficient characteristic kernel matrix;
then, combining the three characteristic kernel function matrixes by using a combined kernel method to obtain a training data set and a test data set of the three types of MSTAR data;
then, respectively substituting the training data sets of the MSTAR three types of data into an expression of the optimal solution of the Bayes multi-core learning support vector machine and an expression of the hidden variable expectation value, setting the maximum iteration times, and finally obtaining the optimal solution of the Bayes multi-core learning support vector machine;
and finally, according to the obtained optimal solution and a test data set of the MSTAR three types of data, calculating to obtain a target identification label.
The results of the identification of the MSTAR class three data by the method of the invention were compared with the results of the other 5 methods, as shown in table 2.
TABLE 2 comparison of results of the inventive method and other methods on the MSTAR three types of data
As can be seen from table 2: the recognition rate of the Bayesian multi-core learning support vector machine model on the SAR image three-class target is 99.12%, compared with the results of other methods, the result is obviously improved, and the performance of the SAR image target recognition is obviously improved by the method.