{"id":19543189,"url":"https://github.com/yeonghyeon/mlp-mixer-tf2","last_synced_at":"2025-06-11T01:08:29.271Z","repository":{"id":159175984,"uuid":"481957839","full_name":"YeongHyeon/MLP-Mixer-TF2","owner":"YeongHyeon","description":"TensorFlow implementation of \"MLP-Mixer: An all-MLP Architecture for Vision\"","archived":false,"fork":false,"pushed_at":"2022-12-20T17:14:01.000Z","size":857,"stargazers_count":5,"open_issues_count":0,"forks_count":0,"subscribers_count":1,"default_branch":"master","last_synced_at":"2025-04-26T17:47:17.089Z","etag":null,"topics":["mixer","mlp","mlp-classifier","mlp-mixer","mnist","mnist-classification","mnist-dataset","multi-layer-perceptron","transformer"],"latest_commit_sha":null,"homepage":"","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/YeongHyeon.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null,"zenodo":null}},"created_at":"2022-04-15T12:54:01.000Z","updated_at":"2023-02-19T15:45:01.000Z","dependencies_parsed_at":null,"dependency_job_id":"c3b441be-9914-44d7-b9cc-83723eeb92c8","html_url":"https://github.com/YeongHyeon/MLP-Mixer-TF2","commit_stats":null,"previous_names":[],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/YeongHyeon%2FMLP-Mixer-TF2","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/YeongHyeon%2FMLP-Mixer-TF2/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/YeongHyeon%2FMLP-Mixer-TF2/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/YeongHyeon%2FMLP-Mixer-TF2/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/YeongHyeon","download_url":"https://codeload.github.com/YeongHyeon/MLP-Mixer-TF2/tar.gz/refs/heads/master","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/YeongHyeon%2FMLP-Mixer-TF2/sbom","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":259178496,"owners_count":22817387,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["mixer","mlp","mlp-classifier","mlp-mixer","mnist","mnist-classification","mnist-dataset","multi-layer-perceptron","transformer"],"created_at":"2024-11-11T03:17:50.678Z","updated_at":"2025-06-11T01:08:29.263Z","avatar_url":"https://github.com/YeongHyeon.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"[TensorFlow 2] MLP-Mixer: An all-MLP Architecture for Vision\n=====\nTensorFlow implementation of \"MLP-Mixer: An all-MLP Architecture for Vision\"\n\n## Concept\n\u003cdiv align=\"center\"\u003e\n  \u003cimg src=\"./figures/mixer.png\" width=\"800\"\u003e  \n  \u003cp\u003eThe MLP-Mixer architecture [1].\u003c/p\u003e\n\u003c/div\u003e\n\n### Patch Embedding\n\u003cdiv align=\"center\"\u003e\n  \u003cimg src=\"./figures/patch_embedding.png\" width=\"800\"\u003e  \n\u003c/div\u003e\n\n### Mixer Block\n\u003cdiv align=\"center\"\u003e\n  \u003cimg src=\"./figures/mixer_layer.png\" width=\"800\"\u003e  \n\u003c/div\u003e\n\n### MLP in Mixer Blcok\n\u003cdiv align=\"center\"\u003e\n  \u003cimg src=\"./figures/mlp_block.png\" width=\"700\"\u003e  \n\u003c/div\u003e\n\n## Results\n\n### Loss\n\n\u003cimg src=\"./figures/Mixer_loss_mean.svg\" width=\"800\"\u003e  \n\n### Performance\n\n|Indicator|Value|\n|:---|:---:|\n|Accuracy|0.97860|\n|Precision|0.97882|\n|Recall|0.97808|\n|F1-Score|0.97829|\n\n```\nConfusion Matrix\n[[ 968    0    0    0    2    0    3    0    4    3]\n [   0 1127    0    1    1    0    2    1    3    0]\n [   3    0 1002    6    6    0    1    5    9    0]\n [   1    0    2  993    0    0    0    2   10    2]\n [   0    1    0    0  974    0    3    0    1    3]\n [   3    0    0   17    4  839    9    2   12    6]\n [   4    2    0    0   10    0  940    0    2    0]\n [   0    3    4    4    5    0    0 1005    3    4]\n [   0    0    3    1    5    0    1    3  956    5]\n [   1    3    0    3   14    1    0    4    1  982]]\nClass-0 | Precision: 0.98776, Recall: 0.98776, F1-Score: 0.98776\nClass-1 | Precision: 0.99208, Recall: 0.99295, F1-Score: 0.99251\nClass-2 | Precision: 0.99110, Recall: 0.97093, F1-Score: 0.98091\nClass-3 | Precision: 0.96878, Recall: 0.98317, F1-Score: 0.97592\nClass-4 | Precision: 0.95397, Recall: 0.99185, F1-Score: 0.97254\nClass-5 | Precision: 0.99881, Recall: 0.94058, F1-Score: 0.96882\nClass-6 | Precision: 0.98019, Recall: 0.98121, F1-Score: 0.98070\nClass-7 | Precision: 0.98337, Recall: 0.97763, F1-Score: 0.98049\nClass-8 | Precision: 0.95504, Recall: 0.98152, F1-Score: 0.96810\nClass-9 | Precision: 0.97711, Recall: 0.97324, F1-Score: 0.97517\n\nTotal | Accuracy: 0.97860, Precision: 0.97882, Recall: 0.97808, F1-Score: 0.97829\n```\n\n## Requirements\n* Tensorflow 2.4.0  \n* \u003ca href=\"https://github.com/YeongHyeon/white-box-layer\"\u003ewhiteboxlayer\u003c/a\u003e 0.2.8\n\n## Reference\n[1] Tolstikhin, Ilya O., et al. \u003ca href=\"https://proceedings.neurips.cc/paper/2021/file/cba0a4ee5ccd02fda0fe3f9a3e7b89fe-Paper.pdf\"\u003e\"Mlp-mixer: An all-mlp architecture for vision.\"\u003c/a\u003e Advances in Neural Information Processing Systems 34 (2021).\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fyeonghyeon%2Fmlp-mixer-tf2","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fyeonghyeon%2Fmlp-mixer-tf2","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fyeonghyeon%2Fmlp-mixer-tf2/lists"}