researchMistral AI

Mistral AI fine-tunes Pixtral-12B on satellite imagery, boosting classification accuracy from 56% to 91%

TL;DR

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.

2 min read
0

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

research

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.

model release

Mistral releases Leanstral 1.5: 119B parameter open-source model for Lean 4 proof assistance

Mistral AI has released Leanstral 1.5, an open-source 119B parameter mixture-of-experts model designed specifically for Lean 4 proof assistance. The model features 128 experts with 4 active per token (6.5B activated parameters), a 256k token context window, and multimodal input capabilities.

research

AI2 Releases DiScoFormer: Single Transformer Estimates Density and Score Across Distributions Without Retraining

Allen Institute for AI (AI2) has released DiScoFormer, a transformer model that estimates both the density and score of any distribution from a sample in a single forward pass without retraining. In 100 dimensions, the model reduces score estimation error by 6.5x and density error by 37x compared to classical kernel density estimation.

research

AI2 Research: Hybrid Models Excel at Content Words, Transformers Better at Token Repetition

Allen Institute for AI researchers conducted token-level analysis comparing their 7B-parameter Olmo 3 transformer and Olmo Hybrid models. The study finds hybrid architectures show a loss gap advantage of 0.04 on content words (nouns, verbs, adjectives) versus 0.02 on function words, while transformers match or exceed hybrids on repeated tokens and closing braces.

Comments

Loading...