CoolFace
Modelpublic

LoliRimuru/AAL-Plus_Image_Quality_Assessment

sourceHugging Faceapache-2.0updated 9mo agoView on Hugging Face
2likes
Model Card

AAL-Plus Image Quality Assessment

A lightweight model that predicts compression quality levels and detects artifacts across multiple image formats (JPEG, WebP, AVIF, JXL). Trained to identify subtle compression artifacts that indicate the quality level used during image encoding.

๐Ÿ“Š Model Performance

Based on validation results from the final training epochs:

MetricValue
Overall Validation Accuracy97.1%
Overall Validation Loss0.0006
Training Accuracy95.6%

Per-Format Accuracy

FormatValidation AccTraining AccQuality Range
JPEG99.4%98.9%0-100
WebP97.0%95.5%0-100
AVIF97.1%95.3%0-100
JXL94.8%92.6%0-100

Accuracy measured as predictions within ยฑ5% range of actual quality values

๐ŸŽฏ Key Features

  • โ€”Multi-format Support: Detects artifacts in JPEG, WebP, AVIF, and JXL formats
  • โ€”Lightweight Architecture: Only ~2M parameters (~8MB model size)
  • โ€”High Precision: 99.4% on JPEG
  • โ€”Fast Inference: Optimized for real-time processing even on very weak HW using either CPU or GPU
  • โ€”Format-agnostic: Single model handles all compression types

Limitations

  1. 1.Resolution Dependency: Model trained on 512x512 crops; very small images may yield reduced accuracy
  2. 2.Quality Range: Model assumes standard quality ranges; non-standard encoders may produce different results
  3. 3.Image Resizing: Trying to detect compression artifacs in images which were compressed and than converted to lossless format and resized may fail or be much less accurate as these artifacts are washed by the resizing algorithms.

Environmental Impact

  • โ€”Training: ~12 GPU hours on RTX 5090
  • โ€”Model Size: 8MB (reduces storage and bandwidth costs)