This is a research project conducted as the final project for COSE474 Deep Learning.
Many real-world problems consist of multiple classes, and there are areas like healthcare where accurate class assignment becomes critical. In general, multi-class classification problems are less accurate than binary classification problems because it becomes more difficult to accurately learn and distinguish the boundaries between each class. This phenomenon becomes more pronounced as the number of classes that need to be distinguished increases, so finding a way to address it is the motivation for this research. Consequently, this study propose a novel model structure aimed at both reducing the complexity of distinguishing classes and enhancing accuracy through a mechanism that compensates for incorrect classifications.
$ python main.py
$ python inference.py
- Before training, please configure the experimental settings in the
experimental_setting.json
file. - The training logs are saved using the TensorBoard library. Please check the storage path.