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