Skip to main content
. 2022 Mar 28;24(4):471. doi: 10.3390/e24040471
Algorithm 1 FW-RFE
Input: The set of output features of the pooling layer and sample labels,
(Pi,yi)(i=1)Num_l,Pip1,p2,,pjj=1Num_f,yi+1,0
   and the set of convolutional layer feature weights,
w11,w21,w101,w12,w22,w102,,w1j,w2j,w10j(j=1)Num_f
Output: Feature Sorted Set R;
  • 1:
    Initial feature set S
    S=siinum_l,si=p1i,p2i,,pji
    and accuracy set V=;
  • 2:

    while(Sϕ)do

  • 3:

        Calculate the ranking criterion score:ci=n=110(wnj)2;

  • 4:
        Find the feature with the smallest ranking score:
    h=minC
  • 5:

        Extract features si=siphi, updata S;

  • 6:

        Take S as the input of the fully connected layer, use sigmoid regression function to achieve binary classification: sigmoid(Si)=11+eSi

  • 7:

        Update accuracy set: V=acc,V

  • 8:

    end while

  • 9:

    When the accuracy rate in the accuracy set V is the highest, the feature set in S is selected as the output result