Apple researchers have released Pico-Banana-400K, a comprehensive dataset containing 400,000 curated images designed to improve how artificial intelligence systems edit photos based on text prompts, the company announced in a research paper published this week.
The massive dataset aims to address what Apple describes as a critical gap in current AI image editing training, where progress has been constrained by inadequate datasets built from real photographs. While systems like GPT-4o can make impressive edits, researchers say the lack of large-scale, high-quality training data has limited advancement in the field.
Systematic Approach to Quality and Diversity
What distinguishes Pico-Banana-400K from previous datasets is Apple’s systematic approach to quality control and comprehensive coverage. The images are organized into 35 different edit types across eight categories, ranging from basic adjustments like color changes to complex transformations such as converting people into Pixar-style characters or LEGO figures.
Apple built the dataset using Google’s Gemini-2.5-Flash-Image model, also known as Nano-Banana, to generate the edits, while Gemini-2.5-Pro served as an automated quality control system to evaluate results based on instruction compliance and technical quality. Each image in the set underwent this rigorous AI-powered screening process before inclusion.
The dataset 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.
Revealing Current AI Limitations
Apple’s research revealed significant limitations in current image editing models. While global style changes succeeded 93% of the time, precise tasks like relocating objects or editing text struggled with success rates below 60%. These findings provide valuable insights into where AI image editing still falls short of user expectations.
The complete Pico-Banana-400K dataset is freely available for non-commercial research use on GitHub, allowing developers and researchers to use it for training more capable image editing AI systems. According to the researchers, the dataset establishes “a robust foundation for training and benchmarking the next generation of text-guided image editing models”.
