Feature-Space Oversampling for Addressing Class Imbalance in SAR Ship Classification
This paper proposes two novel feature-space oversampling algorithms, M2m and M2m, which effectively address class imbalance in SAR ship classification by significantly improving F1-scores on OpenSARShip and FuSARShip datasets compared to existing methods.
Original paper licensed under CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). This is an AI-generated explanation of the paper below. It is not written or endorsed by the authors. For technical accuracy, refer to the original paper. Read full disclaimer
The Big Problem: The "Unfair Classroom"
Imagine you are a teacher trying to teach a class about different types of ships. You have a huge textbook (the dataset) with pictures of ships. However, the book is unbalanced:
- There are 1,000 pictures of big cargo ships.
- There are only 50 pictures of fishing boats.
If you train a student (an AI model) on this book, they will become an expert at spotting cargo ships but will fail miserably at identifying fishing boats. They will just guess "cargo ship" every time because that's what they see most often. In the world of radar (SAR) images, this is a huge problem because these images are often grainy and noisy, making it hard to teach the AI with just a few examples.
The Old Solution: "Copy and Paste" (and why it failed)
Usually, when teachers don't have enough examples of the rare items, they try to "augment" the data. This is like taking a single photo of a fishing boat, flipping it upside down, making it slightly brighter, or zooming in, and pretending these are new, unique photos.
The authors say this doesn't work well for radar images. Because radar images are so noisy and low-resolution, simply copying and pasting them creates fake-looking images that confuse the AI rather than helping it.
The New Solution: "The Feature Space Bakery"
Instead of trying to make more pictures of the rare ships, the authors decided to work with the essence of the ships.
Imagine the AI has already looked at all the pictures and written a "flavor profile" for each one.
- Cargo Ship Flavor: "Heavy, boxy, long."
- Fishing Boat Flavor: "Small, agile, specific shape."
The authors propose a new method called Feature-Space Oversampling. Instead of baking new cakes (images), they take the "flavor profiles" (mathematical features) of the common ships and mix them with the rare ships to create synthetic flavor profiles.
They created two new recipes (algorithms) to do this mixing:
Recipe M2mf (The "Safe Mixer"):
- This method takes a "flavor profile" from a common ship and gently nudges it toward the "center" of the rare ship's flavor.
- The Safety Check: Before adding this new synthetic flavor to the mix, it checks: "Is this new flavor too similar to the ones we already have?" If it's too similar, it throws it away. This ensures every new example is unique and adds something new to the learning process.
Recipe M2mu (The "Similarity Matcher"):
- This method also nudges common flavors toward the rare ones.
- The Safety Check: Instead of checking for uniqueness, it checks: "Does this new flavor actually look like a real rare ship?" It uses a "similarity score" to make sure the synthetic example isn't drifting too far away from what a real fishing boat actually looks like.
The Experiment: The Taste Test
The authors tested these two recipes on two different "cookbooks" (datasets):
- OpenSARShip: A standard collection of radar ship images.
- FuSARShip: A newer, higher-quality collection.
They used three different "taste testers" (AI models: ViT, VGG16, and ResNet50) to see which recipe worked best.
The Results: What Worked?
- The Old Way Failed: The traditional method of copying and pasting images (called M2morig) actually made the AI perform worse or stayed the same. It confused the AI.
- The New Way Succeeded: Both of the authors' new "flavor mixing" recipes (M2mf and M2mu) significantly improved the AI's ability to spot the rare ships.
- On the FuSARShip dataset, the M2mu recipe was the winner, boosting the AI's accuracy by nearly 9%.
- On the OpenSARShip dataset, the M2mf recipe was slightly better, boosting accuracy by about 5%.
The "Underfitting" Surprise:
One of the taste testers (ResNet50) struggled in almost every scenario. The authors suggest this model was too "deep" and complex for the limited amount of data, causing it to get confused (underfit), whereas the other models (ViT and VGG) handled the new recipes much better.
The Bottom Line
The paper proves that for radar ship classification, it's better to mix the "essence" of the data than to copy the pictures.
By creating synthetic features (the mathematical description of the ship) rather than synthetic images, the AI learns to recognize rare ships much better. This approach is also cheaper and faster because you don't have to re-process the entire image; you just mix the mathematical "flavors" and train the final classifier.
In short: Don't try to fake more photos of rare ships. Instead, mathematically blend the "vibe" of common ships to create new, realistic examples of the rare ones, and your AI will learn much faster.
Drowning in papers in your field?
Get daily digests of the most novel papers matching your research keywords — with technical summaries, in your language.