How To Get Started With Visual AI – The New AutoML Solution By DataRobot

Visual AI DataRobot

DataRobot has gained traction in the AutoML world due to its intuitive platform that can be leveraged to build ML models without the need for data scientists. In an attempt to further enhance the platform, the firm introduced Visual AI in the DataRobot 6.0 to automate ML workflows with computer vision technology.

The company has been making great strides in the data science landscape. It has been committed to continually improve its platform to simplify the workload of data scientists and in turn, bring efficiency within organizations. In December 2019, DataRobot acquired Paxata to enhance its platform’s capabilities after bagging $200 Million in Series E funding. The firm has been aggressively acquiring and integrating new features to streamline AI workflows.

What is Visual AI  

Computer vision technology has become the foundation for many AI-based applications, such as facial recognition, and object detection, among others. Consequently, DataRobot has come up with a new solution to simplify the incorporation of image data into ML models alongside tabular and text-based data types.


Sign up for your weekly dose of what's up in emerging technology.

With this, anyone can build models by just dragging and dropping images into the DataRobot platform. One can also get started with the models by feeding only a few hundred images for training, thereby getting outputs within minutes or hours. Also, you do not require GPUs as it is optimized to perform even on basic hardware. This is because the firm already offers pre-trained neural networks that do the heavy lifting.

How Is It Different?

The Visual AI solution empowers users to build binary and multiclass classification and regression models with images. One can use it to develop completely new image-based models. Besides, users can also add pictures as new features to existing models to enhance their accuracy from totally different images. In other words, the variables can be extended even after the model is trained, resulting in more flexibility in the data science workflows. 

Download our Mobile App

Getting started with the solution:

These solutions are self-explanatory and point to the factors that lead to the outcome of models. The steps are as follows:-

  1. Create a zip file of images: Keep images into different folders or use comma separated values (CSV) file if you want to include additional features; and then zip the files.
  2. Drag and drop the zip file into the platform: Upload or drag and drop the zip file into a new project and pick your target.
  3. Explore your data: The platform will perform exploratory data analysis (EDA) and show interesting statistics, along with missing and duplicate data.
  4. Model training: It will automatically train, test, and compare different models to recommend the best one eventually.
  5. Evaluation: The models can be evaluated for its accuracy using automated visualizations or neural network visualizer, image embeddings and activation maps. The solution shows how the model pre-processed the data, and why a particular algorithm was picked over the other, and where the neural network looked in the image for every single prediction.
  6. Tune and tweak: If needed, users can tune the models by changing the hyperparameters. However, an expert is required to ensure proper values selection for hyperparameters. 
  7. Deploy and monitor: On deploying the model into production, the solutions provide a view where one can monitor, manage, and enhance the performance. 

Performance Of The Solution

The solution was tested with ~14k images of natural scenes such as buildings, forests, streets, mountains, etc. Visual AI trained 40 models in just two hours without the support of GPUs. The accuracy was around 92%, which was tuned to get even higher results.

Although the results were exceptional, one may have to check for bias and approach an expert to interpret the explainability. Consequently, one cannot rely on the solution even though the models are transparent. AutoML has been helping data scientists, but it will be strenuous for non-experts to work with such tools.


Nevertheless, Visual AI delivered results which were higher than one would have envisioned, but the firm cannot guarantee similar results unless it is tested with different types of data. It is undoubtedly a great addition to DataRobot offerings, but might still require experts to make the most of it. But this has been the case with other AutoML solutions – these still need specialists to enhance outputs. One cannot directly use outputs of models from AutoML solutions and make business decisions based on it, especially when experts are critical of the computer vision technology due to its potential bias outcomes.

More Great AIM Stories

Rohit Yadav
Rohit is a technology journalist and technophile who likes to communicate the latest trends around cutting-edge technologies in a way that is straightforward to assimilate. In a nutshell, he is deciphering technology. Email:

AIM Upcoming Events

Regular Passes expire on 3rd Mar

Conference, in-person (Bangalore)
Rising 2023 | Women in Tech Conference
16-17th Mar, 2023

Early Bird Passes expire on 17th Feb

Conference, in-person (Bangalore)
Data Engineering Summit (DES) 2023
27-28th Apr, 2023

Conference, Virtual
Deep Learning DevCon 2023
27 May, 2023

3 Ways to Join our Community

Telegram group

Discover special offers, top stories, upcoming events, and more.

Discord Server

Stay Connected with a larger ecosystem of data science and ML Professionals

Subscribe to our Daily newsletter

Get our daily awesome stories & videos in your inbox

What went wrong with Meta?

Many users have opted out of Facebook and other applications tracking their activities now that they must explicitly ask for permission.