Apple's New AI Dataset Aims to Improve Photo Editing Models - MacRumors
Skip to Content

Apple's New AI Dataset Aims to Improve Photo Editing Models

Apple researchers have released Pico-Banana-400K, a comprehensive dataset of 400,000 curated images that's been specifically designed to improve how AI systems edit photos based on text prompts.

apple intelligence black
The massive dataset aims to address what Apple describes as a gap in current AI image editing training. While systems like GPT-4o can make impressive edits, the researchers say progress has been limited by inadequate training data built from real photographs. Apple's new dataset aims to improve the situation.

Pico-Banana-400K features images organized into 35 different edit types across eight categories, from basic adjustments like color changes to complex transformations such as converting people into Pixar-style characters or LEGO figures. Each image went through Apple's AI-powered quality control system, with Google's Gemini-2.5-Pro being used to evaluate the results based on instruction compliance and technical quality.

The dataset also includes three specialized subsets: 258,000 single-edit examples for basic training, 56,000 preference pairs comparing successful and failed edits, and 72,000 multi-turn sequences showing how images evolve through multiple consecutive edits.

Apple built the dataset using Google's Gemini-2.5-Flash-Image (aka Nano-Banana) editing model, which was released just a few months ago. However, Apple's research revealed its limitations. While global style changes succeeded 93% of the time, precise tasks like relocating objects or editing text seriously struggled, with success rates below 60%.

apple image editing ai dataset pico banana
Despite the limitations, researchers say their aim with Pico-Banana-400K is to establish "a robust foundation for training and benchmarking the next generation of text-guided image editing models." The complete dataset is freely available for non-commercial research use on GitHub, so developers can use it to train more capable image editing AI.

Popular Stories

Apple Event Logo

Apple's Next Era Begins September 1

Thursday May 7, 2026 10:36 am PDT by
Apple recently announced that Tim Cook will be stepping down as CEO later this year, after 15 years of leading the company. Effective September 1, Apple's hardware engineering chief John Ternus will become the company's next CEO, while Cook will become executive chairman of Apple's board of directors. In his new role, Apple said Cook will assist with "certain aspects" of the company,...
Instagram Feature 2

PSA: Instagram Encrypted Messaging Ends on Friday, May 8

Tuesday May 5, 2026 8:24 am PDT by
Instagram will remove end-to-end encryption for direct messages between users from May 8, 2026. When the date comes around, Meta will potentially be able to see the contents of all messages between users on the social media platform. Encrypting messages has been an optional feature in Instagram since 2023, but in March of this year the social media platform quietly updated a help page to say ...
macbook neo launch day

Apple May Drop Base $599 MacBook Neo as Chip, DRAM Costs Climb

Thursday May 7, 2026 4:55 am PDT by
Apple is considering dropping the cheapest MacBook Neo configuration as one possible response to the rising cost of building the popular laptop, according to Taiwan-based tech columnist and former Bloomberg reporter Tim Culpan. The Neo currently starts at $599 for a 256GB model, with a 512GB version at $699. Writing in his latest Culpium newsletter, Culpan says cutting the entry-level...

Top Rated Comments

JSRinUK Avatar
7 months ago
Ah, with Apple’s involvement I enthusiastically await the “Photo Tools” equivalent of this popping up whenever I want to use it…



Attachment Image
Score: 11 Votes (Like | Disagree)
mjs916 Avatar
7 months ago
I realize this isn’t the point but… Can Apple’s own AI tools do what Gemini can? I’ve been unimpressed by Image Playground.
Score: 11 Votes (Like | Disagree)
turbineseaplane Avatar
7 months ago
Ok ... sure 🤷‍♂️

I'm very much in the "show me, don't tell me" camp with Apple and any of their AI / Siri efforts.
Score: 11 Votes (Like | Disagree)
rp2011 Avatar
7 months ago
Anyone who has used any of the AI tools knows none of them are ready for Prime Time. None of them. That is why I always say it's just echo-chamber nonsense to say Apple or anyone is behind in this nacent quickly evolving open source media. That ALL of them have been RUSHED to market without adequate testing goes without saying.

There is immense room for improvement with just the low hanging fruit of better training data. And here we see Apple showing they certainly know that.
Score: 10 Votes (Like | Disagree)
everlast3434 Avatar
7 months ago
Converting people into Legos. Seriously lol. Do people want this stuff?
Score: 8 Votes (Like | Disagree)
Mac Fly (film) Avatar
7 months ago

Anyone who has used any of the AI tools knows none of them are ready for Prime Time. None of them. That is why I always say it's just echo-chamber nonsense to say Apple or anyone is behind in this nacent quickly evolving open source media. That ALL of them have been RUSHED to market without adequate testing goes without saying.

There is immense room for improvement with just the low hanging fruit of better training data. And here we see Apple showing they certainly know that.
100%

All of the LLMs are confidently wrong the whole time.
Score: 6 Votes (Like | Disagree)