Mistral AI fine-tunes Pixtral-12B on satellite imagery, boosting classification accuracy from 56% to 91%
Mistral AI has published research showing that fine-tuning its Pixtral-12B vision language model on satellite imagery increases classification accuracy from 56% to 91% on the Aerial Image Dataset. Using Low-Rank Adaptation (LoRA) with 8,000 training samples across 30 scene categories, the company reduced hallucinations from 5% to 0.1% for under $10 in compute costs.
Mistral AI fine-tunes Pixtral-12B on satellite imagery, boosting classification accuracy from 56% to 91%
Mistral AI has published research demonstrating that fine-tuning its Pixtral-12B vision language model on satellite imagery produces a 1.6x improvement in classification performance. The base model achieved 56% accuracy on the Aerial Image Dataset (AID), while the fine-tuned version reached 91% accuracy.
Technical approach: LoRA fine-tuning
The company used Low-Rank Adaptation (LoRA), a technique that injects small trainable matrices into model weights rather than retraining the entire model. According to Mistral AI, this approach required 8,000 training samples distributed across 30 scene categories from the Aerial Image Dataset, introduced by Xia et al under a Public Domain license.
The fine-tuning job cost under $10 to run, making it accessible for specialized domain adaptation. Mistral AI reports that hallucinations—cases where the model generated invalid class names not in the target set—dropped from 5% to 0.1% after fine-tuning.
Dataset and classification challenges
The Aerial Image Dataset contains satellite imagery classified into detailed scene categories including Desert, BareLand, RailwayStation, Mountain, and 26 other classes. Many categories proved difficult for the base model to distinguish, particularly visually similar classes like "Dense Residential" vs. "Medium Residential" or ambiguous scenes labeled "Center."
Mistral AI's example highlights the model's improved ability to differentiate between "Playground" and "Stadium"—the base model classified both as "Stadium," while the fine-tuned version correctly identified the distinction based on the presence of surrounding seats.
Implementation details
The research used a train/test split of 8,000 and 2,000 samples respectively. According to Mistral AI, minimal hyperparameter tuning was required. The company recommends:
- Starting with small learning rates to avoid overshooting optimal weights
- Beginning with a single training epoch and monitoring for overfitting
- Using batch sizes that fit computational resources while maintaining stable gradients
Fine-tuning can be executed via Mistral's API or through the La Plateforme UI. The API provides direct control over hyperparameters, while La Plateforme automatically computes optimal batch size based on dataset size.
What this means
This research validates that domain-specific fine-tuning of general-purpose vision language models can achieve significant performance gains on specialized imagery tasks. The sub-$10 cost and 8,000-sample requirement makes this approach viable for organizations with proprietary satellite data.
The technique extends beyond satellite imagery to other underrepresented visual domains in standard VLM training sets, including medical image captioning, surveillance footage analysis, and ancient manuscript transcription. Mistral AI has published the implementation in a Jupyter notebook at github.com/mistralai/cookbook.
The results suggest that for tasks requiring nuanced visual distinctions in specialized domains, fine-tuning substantially outperforms prompt engineering approaches, which Mistral AI notes can produce inconsistent results on complex classification tasks.
Related Articles
Google DeepMind's GenCeption uses video generator for computer vision with 500x less training data
Google DeepMind researchers developed GenCeption, which repurposes Alibaba's Wan2.1 video generator for computer vision tasks including depth estimation, segmentation, and 3D pose estimation. The model matches state-of-the-art specialized systems while training on only 7,500 synthetic videos—between 7 and 500 times less data than competing approaches.
AWS introduces rDPO unlearning technique to reduce false content moderation in Amazon Nova models by 53 percentage point
AWS has developed Reverse Direct Preference Optimization (rDPO), a novel unlearning technique that reduces over-deflection in Amazon Nova models by up to 53 percentage points. The approach allows organizations to selectively adjust content moderation safeguards while preserving general model capabilities through LoRA adapters.
Meta AI Pairs a Second 'Memory Agent' With Coding Agents, Lifts Terminal-Bench Score From 38% to 46%
Meta AI researchers describe a plug-in 'memory agent' that runs alongside an unmodified 'action agent,' deciding when to inject reminders about past constraints and failures. The system lifted Terminal-Bench 2.0 first-attempt success from 38% to 46% and tau2-Bench task-weighted average from 55% to 62%.
Black Forest Labs Reports 10x Fewer Safety Vulnerabilities Than Competitors in FLUX.2 Model Family
Black Forest Labs reports its FLUX.2 image generation models demonstrate more than 10 times fewer vulnerabilities for synthetic non-consensual intimate imagery (NCII) and child sexual abuse material (CSAM) compared to other leading open-weight models. The company claims targeted post-training mitigations reduced vulnerabilities by 77-98% before release, according to third-party red-teaming conducted by Cinder.
Comments
Loading...