OSLNet

OSLNet enhances classification of small-sample datasets by introducing an Orthogonal Softmax Layer (OSL) that enforces orthogonal weight vectors to improve discriminative feature learning and reduce overfitting.


Key Features:

  • Orthogonal Softmax Layer (OSL): Enforces orthogonality of weight vectors in the classification layer during training and testing to reduce feature redundancy and enlarge decision margins.
  • Reduced Rademacher Complexity: Lowers the network Rademacher complexity to 1/K (where K is the number of classes) compared to traditional fully connected classification layers, yielding tighter generalization error bounds.
  • Enhanced Generalization: Improves generalization for small-sample datasets and demonstrates superior performance compared to existing methods across benchmark datasets.

Scientific Applications:

  • Biomedical Data Analysis: Applicable to genomics and proteomics classification tasks where large datasets are difficult to obtain.
  • Rare Disease Diagnosis: Suited for classification problems involving limited patient samples common in rare disease studies.

Methodology:

Constructs a deep neural network with multiple nonlinear layers and incorporates an Orthogonal Softmax Layer that maintains orthogonal weight vectors during training and testing to reduce overfitting and improve generalization.

Topics

Details

License:
MIT
Tool Type:
command-line tool
Programming Languages:
Python
Added:
1/18/2021
Last Updated:
3/15/2021

Operations

Publications

Li X, Chang D, Ma Z, Tan Z, Xue J, Cao J, Yu J, Guo J. OSLNet: Deep Small-Sample Classification With an Orthogonal Softmax Layer. IEEE Transactions on Image Processing. 2020;29:6482-6495. doi:10.1109/tip.2020.2990277. PMID:32386152.

PMID: 32386152
Funding: - National Key Research and Development Program of China: 2018YFB2100500, 2019YFF0303300, 2019YFF0303302 - National Natural Science Foundation of China: 61763028, 61773071, 61906080, 61922015, 61976138, 61977047, U19B2036 - National Science and Technology Major Program of the Ministry of Science and Technology: 2018ZX03001031 - Beijing Academy of Artificial Intelligence: BAAI2020ZJ0204 - Beijing Nova Programme Interdisciplinary Cooperation Project: Z191100001119140 - Key Program of Beijing Municipal Natural Science Foundation: L172030 - Shanghai Science and Technology Committee: 2015F0203-000-06 - Shanghai Municipal Education Commission: 2019-01-07-00-01-E00003