Lohan11's picture
Initial release: technical documentation, quantization scripts, and NPU benchmarks
73072fb verified
|
Raw History Blame Contribute Delete
2.88 kB

Model Card: Lithotheque Edge AI Models

Model Details

  • Developer: GeoStratum
  • Model Versions: VisionAiManager (MobileNetV5-300m / MobileNetV4-Large) and LocalAiManager (Metrology).
  • Model Type: Convolutional Neural Networks (CNN) optimized for Edge Inference (INT8/INT4 Quantization).
  • License: Proprietary (GeoStratum).
  • Training Data: Hybrid corpus of proprietary metrology images and open-source MIT-licensed geological datasets (Udayl, Stealth Technologies).

Intended Use

  • Primary Use Case: Offline, on-device geological and lithological classification for field geologists, students, and enthusiasts.
  • Secondary Use Case: Scale detection and topographical metrology mapping via reference objects.
  • Target Environment: Mobile devices operating in remote areas without internet connectivity.

Out-of-Scope Use

These models are designed for educational and preliminary field classification. They MUST NOT be used for:

  • Critical Engineering Decisions: Assessing rock mechanics, stability, or load-bearing capacities for civil engineering or construction.
  • Financial/Mining Exploration: Making high-stakes financial decisions regarding resource extraction or ore grade estimation.
  • Safety: Detecting hazardous environments or toxic minerals for safety compliance. Always consult laboratory analysis (e.g., XRD, XRF) for definitive geological validation.

Metrics & Performance

  • Accuracy (Standardized Multi-Source Benchmark):
    • MobileNetV5-300M (Premium / Standard Tier): Up to 88.33% Top-1 / 92.67% Top-3 in baseline FP16/BF16, 83.33% Top-1 / 89.33% Top-3 in INT8 (W8A8), and 80.33% Top-1 / 87.67% Top-3 in W4A8 Full Integer.
    • MobileNetV4-Large (Balanced / Legacy Tier): 78.78% Top-1 / 91.37% Top-3 in baseline FP16/BF16, 77.84% Top-1 / 90.42% Top-3 in INT8, and 73.91% Top-1 / 87.15% Top-3 in W4A8 (reaching >91% Top-1 on in-situ daylight field subsets).
  • Latency: Sub-3 ms on dedicated NPU hardware for MobileNetV4-Large (~400 FPS steady-state), and 12–18 ms for 1024×1024 Scale Detection.
  • For comprehensive hardware profiling, please refer to the main README.md.

Limitations & Bias

  • Occlusion & Surface Conditions: Accuracy degrades if the rock sample is heavily covered in mud, moss, or artificial coatings. Freshly broken surfaces yield the most accurate predictions.
  • Lighting Dependency: Extreme low-light conditions or harsh direct camera flash may alter texture perception, slightly increasing the margin of error.
  • Micro-crystalline vs. Cryptocrystalline: Some visually identical rocks require chemical or microscopic analysis for definitive differentiation. The model predicts the highest mathematical probability based strictly on visual morphological features.