Skip to content

GECCO is a lightweight image classifier based on single MLP and graph convolutional layers. We find that our model can achieve up to 16x better latency than other state-of-the-art models. The paper for our model can be found at https://arxiv.org/abs/2402.00564

Notifications You must be signed in to change notification settings

GECCOProject/GECCO

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GECCO

About

GECCO is a lightweight image classifier based on single MLP and graph convolutional layers. We find that our model can achieve up to 16x better latency than other state-of-the-art models. The paper for our model can be found at https://arxiv.org/abs/2402.00564

Reproducibility

To reproduce the results in the GECCO paper, the practitioner should change the featurelength (hidden size) of the model in the model.py file. For MNIST, MSTAR, use featurelength of 32, MSTAR, use a featurelength of 48, and CXR, use a featurelength of 56. Additionally, update the directories in the code to match the datasets in your own workspace.

Citing

If you use GECCO in your paper, please use the following BibTeX entry.

@misc{feinashley2024singlegraphconvolutionneed,
      title={A Single Graph Convolution Is All You Need: Efficient Grayscale Image Classification}, 
      author={Jacob Fein-Ashley and Sachini Wickramasinghe and Bingyi Zhang and Rajgopal Kannan and Viktor Prasanna},
      year={2024},
      eprint={2402.00564},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2402.00564}, 
}

About

GECCO is a lightweight image classifier based on single MLP and graph convolutional layers. We find that our model can achieve up to 16x better latency than other state-of-the-art models. The paper for our model can be found at https://arxiv.org/abs/2402.00564

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages