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,...
Four iPhone 18 Pro Colors Mock Feature

iPhone 18 Pro Launching in September With These 10 New Features

Saturday May 9, 2026 6:03 am PDT by
While the iPhone 18 Pro and iPhone 18 Pro Max are not launching until September, there are already plenty of rumors about the devices. It was initially reported that the iPhone 18 Pro models would have fully under-screen Face ID, with only a front camera visible in the top-left corner of the screen. However, the latest rumors indicate that only one Face ID component will be moved under the...
Apple Watch Series 10 Jet Black Touch ID

Apple Watch Series 12 and watchOS 27: What to Expect Later This Year

Friday May 8, 2026 2:07 pm PDT by
While not too much has been reported about the next Apple Watch models, there are a few rumors about potential design changes and watchOS 27 features. Apple Watch Series 12 and Apple Watch Ultra 4 models are expected to be released in September, and we have outlined some of the key rumored hardware and software changes below. A new Apple Watch SE is not expected this year, as that model was...

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)