How to Train an Image Classification Model from Labeled Images in Cloud Storage?

PMLE By islever Editorial Last updated

Short answer

C Import the labeled images as a managed dataset in Vertex AI and use AutoML to train the model.

Question

You need to develop an image classification model by using a large dataset that contains labeled images in a Cloud Storage bucket. What should you do?
  1. Use Vertex AI Pipelines with the Kubeflow Pipelines SDK to create a pipeline that reads the images from Cloud Storage and trains the model.
  2. Use Vertex AI Pipelines with TensorFlow Extended (TFX) to create a pipeline that reads the images from Cloud Storage and trains the model.
  3. Import the labeled images as a managed dataset in Vertex AI and use AutoML to train the model. Correct Answer
  4. Convert the image dataset to a tabular format using Dataflow Load the data into BigQuery and use BigQuery ML to train the model.

Key Takeaways

Tests knowledge of Vertex AI AutoML for ready-to-use model training, with the common trap being over-engineering by choosing complex ML pipelines like TFX when a managed service suffices.

Train image classification models efficiently on Google Cloud using Vertex AI AutoML with labeled images stored in Cloud Storage, as confirmed by community consensus favoring simplicity over custom pipeline development.

Common mistake: Option B is frequently selected due to TensorFlow Extended's robustness for production workflows, but it introduces unnecessary infrastructure overhead when the scenario only requires straightforward model training from labeled data.

Expert Analysis

Why the Answer Is Correct

Vertex AI AutoML is specifically engineered to train high-quality machine learning models without requiring custom code or orchestration frameworks. It natively supports importing labeled image datasets directly from Cloud Storage buckets into managed datasets, automatically handling preprocessing, training, and evaluation for image classification tasks.

Why the Other Options Are Wrong

Options A and B mandate building full ML pipelines using Kubeflow or TFX, which are designed for custom training loops, advanced feature engineering, and production deployment rather than quick managed training. Option D incorrectly suggests converting image files into tabular formats for BigQuery ML, which cannot process raw pixel data required for visual recognition. These alternatives violate the principle of least effort when the objective is strictly model development.

Community Comment Notes

The majority of candidates correctly identify C as the most direct solution, emphasizing that AutoML removes pipeline complexity when the goal is purely training a model. Several users initially debated B, citing TFX's integration capabilities, but acknowledged that the prompt lacks requirements for custom scripts or automated MLOps workflows. Commenters also referenced official documentation to verify that managed datasets seamlessly ingest labeled images from GCS buckets.

Official Reference

https://cloud.google.com/vertex-ai/docs/tutorials/image-classification-automl/dataset

This link opens the official vendor documentation in a new tab.

Exam Strategy

Match the scenario's primary objective to the simplest viable Google Cloud service; avoid selecting complex orchestration tools like Dataflow or Kubeflow unless the question explicitly demands custom training logic, hyperparameter tuning, or production pipeline automation.

Community Votes

C
65%
B
35%

65% of anonymous learners picked answer C. Votes are pick records left by other test-takers — they are not the verified answer.

Discussion 9 contributions

Compiled from public community threads and reader submissions, lightly edited for clarity. Handles and vote tallies are shown as recorded, and comments reflect the views of individual contributors.

From public community threads 9

  1. guilhermebutzke 👍 4 Answered B
    My answer: B TensorFlow Extended (TFX) and Kubeflow provide capabilities for building machine learning pipelines that can handle data stored in Google Cloud Storage (GCS). However, when it comes to ease of use specifically for working with data in GCS, TFX may have a slight edge over Kubeflow for 1- Integration with GCS- TensorFlow: TFX is tightly integrated with TensorFlow that has built-in support for GCS and provides convenient APIs for reading data directly from GCS buckets 2 - Abstraction of Data Handling TFX provides higher-level abstractions and components specifically designed for common machine learning tasks, including data preprocessing, model training, and model evaluation
  2. b1a8fae 👍 2 Answered C
    Very vaguely put. I choose C over B just because it sounds like a simpler approach, but both should theoretically work.
  3. f084277 👍 2 Answered C
    You're just trying to TRAIN A MODEL, not set up a whole pipeline. Answer is clearly C
  4. AzureDP900 👍 1
    B is right in my opinion, while both options C and B involve importing labeled images into Vertex AI, using AutoML for image classification might not be the most suitable choice. TFX is a more specialized tool that provides a robust pipeline framework specifically designed for image classification tasks, making it a better fit for this particular use case.
  5. sekhrivijay 👍 1 Answered B
    Managed dataset has a size limitation of 100GB . Question states " a large dataset " . Unmanged dataset has not size limitation . Assuming large here implies > 100GB , it should eliminate answer C
Show 4 more comments
  1. NamitSehgal 👍 1 Answered C
    leverage Vertex AI's AutoML capabilities to automatically build a high-quality image classification model
  2. BlehMaks 👍 1 Answered A
    95th is the similar question. https://cloud.google.com/vertex-ai/docs/pipelines/build-pipeline#sdk
  3. pinimichele01 👍 1 Answered C
    https://cloud.google.com/vertex-ai/docs/tutorials/image-classification-automl/dataset
  4. winston9 👍 3 Answered C
    It's C

Reader comments & corrections

No reader corrections yet — spotted an error or have a note? Share it below.

Log in to comment, report an error, or add a note about this question.

How to Minimize Development Time for Weekly TensorFlow Retraining in Vertex AI?

S-grade

How to Handle Categorical Features in BigQuery ML for Fast Deployment?

S-grade

How to preprocess BigQuery data for TensorFlow?

S-grade

How to Track Model and Prediction Lineage in Vertex AI?

S-grade

Practice All PMLE Questions

Access 65 questions with complete answers and detailed explanations.

View Full PMLE Practice Test →

← Back to PMLE Study Guide