Instructions to use me-aas/EinsteinNet with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use me-aas/EinsteinNet with Keras:
# Available backend options are: "jax", "torch", "tensorflow". import os os.environ["KERAS_BACKEND"] = "jax" import keras model = keras.saving.load_model("hf://me-aas/EinsteinNet") - Notebooks
- Google Colab
- Kaggle
Create README.md
Browse files# EinsteinNet: Physics-Informed Neural Network for Image Classification
EinsteinNet is a high-performance deep learning model designed for advanced image classification. It utilizes a physics-informed architecture to improve feature extraction and structural generalization in scientific and engineering datasets.
## Model Details
- **Developed by:** [Ashif Ahmed Shuvo](https://huggingface.co/me-aas)
- **Model type:** Convolutional Neural Network (CNN) / Physics-Informed
- **Framework:** Keras / TensorFlow
- **Original Source:** [Kaggle EinsteinNet (v1.0.6)](https://www.kaggle.com/models/ashifahmedshuvo/einsteinnet)
- **License:** MIT
## Evaluation Results
According to the performance analysis detailed in *["EinsteinNet and state-of-the-art ML models for android-based orange classification: Integration, evaluation, and deployment"](https://doi.org/10.1016/j.atech.2025.101072)* (2025), the model achieves the following benchmarks on scientific imaging tasks:
| Metric | **EinsteinNet** | Google Teachable Machine | ResNet50 | DenseNet121 | MobileNetV2 | NASNetMobile |
| :--- | :--- | :--- | :--- | :--- | :--- | :--- |
| **Accuracy** | **0.996** | 0.9987 | 0.8377 | 0.9986 | 0.9977 | 0.9986 |
| **Precision (Macro)** | **0.996** | 0.9987 | 0.8389 | 0.9986 | 0.9977 | 0.9986 |
| **Recall (Macro)** | **0.996** | 0.9987 | 0.8377 | 0.9986 | 0.9977 | 0.9986 |
| **F1-Score (Macro)** | **0.996** | 0.9987 | 0.8369 | 0.9986 | 0.9977 | 0.9986 |
| **Trainable Parameters** | **207,109** | N/A | 262,917 | 131,845 | 164,613 | 135,941 |
| **Training Time (s)** | **13,033.01** | 821 | 1,903.58 | 985.11 | 334.29 | 1,061.34 |
| **Model Size (MB)** | **2.54** | 2.34 | 90.98 | 27.34 | 9.24 | 16.81 |
| **Quantized Size (KB)** | **254** | 665 | 23,931 | 7,305 | 2,807 | 5,340 |
### Key Takeaways
* **Efficiency:** EinsteinNet maintains a very high accuracy ($>99\%$) while remaining extremely lightweight ($2.54$ MB), making it significantly more efficient for deployment than ResNet50 ($90.98$ MB).
* **Optimization:** The quantized version of EinsteinNet is only $254$ KB, specifically optimized for edge computing and low-power devices.
For full experimental details, refer to the paper: [DOI: 10.1016/j.atech.2025.101072](https://doi.org/10.1016/j.atech.2025.101072).
## Intended Use
- **Research:** Investigating physics-informed machine learning (PIML) applications.
- **Classification:** High-accuracy labeling for thermal, biological, or structural imaging.
- **Education:** Demonstrating Keras-based CNN implementations for specialized domains.
## How to Load
You can use the `huggingface_hub` library to load this Keras model:
```python
from huggingface_hub import from_pretrained_keras
model = from_pretrained_keras("me-aas/EinsteinNet")
# Example prediction
# predictions = model.predict(input_data)
```
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
metrics:
|
| 6 |
+
- accuracy
|
| 7 |
+
- precision
|
| 8 |
+
- recall
|
| 9 |
+
- f1
|
| 10 |
+
base_model:
|
| 11 |
+
- me-aas/EinsteinNet
|
| 12 |
+
pipeline_tag: image-classification
|
| 13 |
+
library_name: keras
|
| 14 |
+
tags:
|
| 15 |
+
- Research
|
| 16 |
+
- Education
|
| 17 |
+
- Science and Technology
|
| 18 |
+
- Artificial Intelligence
|
| 19 |
+
- Computer Science
|
| 20 |
+
- Computer Vision
|
| 21 |
+
- CNN
|
| 22 |
+
- Image
|
| 23 |
+
- Keras
|
| 24 |
+
- TensorFlow
|
| 25 |
+
- Python
|
| 26 |
+
---
|