AI0-001 AI Models and Data Engineering Practice Test 1 — 15 Questions

Question 1

A data scientist is preparing a dataset for training a classification model. The dataset contains 10,000 records with a binary target variable where 9,500 belong to class A and 500 belong to class B. Which technique should the scientist use to address the class imbalance?

Accepted Answer

SMOTE (Synthetic Minority Oversampling Technique). SMOTE is the correct technique because it generates synthetic samples for the minority class (class B) by interpolating between existing minority instances, effectively balancing the dataset without losing information. This approach avoids the overfitting risk of simple oversampling and the information loss of undersampling, making it ideal for a 19:1 imbalance ratio.

Answer

Random undersampling of class A

Answer

Adding Gaussian noise to class B

Answer

Principal Component Analysis (PCA)

Question 2

An engineer is building a regression model to predict housing prices. The dataset includes features such as square footage, number of bedrooms, and year built. The engineer notices that the square footage values range from 500 to 10,000, while the number of bedrooms ranges from 1 to 5. Which preprocessing step is most critical before training a gradient descent-based model?

Accepted Answer

Normalize or standardize the features. Gradient descent-based models are sensitive to the scale of input features because they update weights proportionally to the gradient, which is influenced by feature magnitudes. With square footage ranging 500–10,000 and bedrooms 1–5, the larger feature will dominate the gradient, causing slow or unstable convergence. Normalizing or standardizing (e.g., Z-score or min-max scaling) ensures all features contribute equally, leading to faster and more reliable training.

Answer

Use k-fold cross-validation

Answer

Apply log transformation to all features

Answer

One-hot encode the features

Question 3

A machine learning team is deploying a sentiment analysis model for customer reviews. The model was trained on reviews from an e-commerce site but will be used for a social media platform. The team observes a drop in accuracy. Which concept best explains this issue?

Accepted Answer

Data drift. Data drift occurs when the statistical properties of the input data change between the training and production environments. Here, the model was trained on e-commerce reviews but is now processing social media posts, which have different vocabulary, tone, and structure, causing a mismatch in the input distribution and leading to accuracy degradation.

Answer

Concept drift

Answer

Bias-variance tradeoff

Answer

Overfitting

Question 4

A data engineer needs to design a data pipeline for a real-time fraud detection system. The system requires low-latency processing of streaming transactions. Which architecture is most appropriate?

Accepted Answer

Stream processing with Apache Kafka and Flink. Apache Kafka provides a distributed, fault-tolerant event streaming platform that ingests high-throughput transaction data with low latency, while Apache Flink offers true stream processing with exactly-once semantics and sub-second event-time processing. Together, they enable real-time fraud detection by analyzing transactions as they arrive, without the delays inherent in batch or micro-batch approaches.

Answer

Data lake with Apache Spark

Answer

Batch processing with Apache Hadoop

Answer

Microservices architecture with REST APIs

Question 5

A team is training a deep learning model for image classification. The training loss decreases rapidly but validation loss starts increasing after a few epochs. Which regularization technique should be applied to mitigate this issue?

Accepted Answer

Early stopping. Option C is correct because early stopping halts training when validation loss increases, preventing overfitting. Option A is wrong because L2 regularization penalizes large weights but doesn't stop training. Option B is wrong because dropout randomly drops neurons during training, but early stopping directly addresses the symptom. Option D is wrong because data augmentation increases data diversity, but the issue is overfitting due to training too long.

Answer

Data augmentation

Answer

L2 regularization

Answer

Dropout

Question 6

An organization needs to store sensitive customer data for training a machine learning model. The data must be encrypted at rest and in transit, and access must be audited. Which combination of practices should be implemented?

Accepted Answer

Use TLS for transfer, AES-256 for storage, and AWS CloudTrail for auditing. Option A is correct because it combines TLS (Transport Layer Security) for encrypting data in transit, AES-256 for strong encryption at rest, and AWS CloudTrail for auditing API-level access. TLS ensures confidentiality and integrity during transmission, AES-256 provides robust symmetric encryption for stored data, and CloudTrail logs all AWS API calls for compliance and audit trails. This triad satisfies the requirements of encryption in transit, at rest, and audited access.

Answer

Use FTP for transfer, AES-128 for storage, and manual log review

Answer

Use SSH for transfer, store data in a database, and enable access logs

Answer

Use MD5 for hashing, store data in plaintext, and enable server logs

Question 7

A data analyst is cleaning a dataset and finds that 20% of the values for the 'age' column are missing. Which imputation method is most robust if the data is not normally distributed?

Accepted Answer

Median imputation. Median imputation is the most robust method for handling missing values in the 'age' column when the data is not normally distributed because the median is unaffected by outliers or skewness. Unlike the mean, which is sensitive to extreme values, the median provides a central tendency measure that better represents the typical value in non-normal distributions, preserving the dataset's integrity for downstream modeling.

Answer

Mean imputation

Answer

Mode imputation

Answer

Remove rows with missing values

Question 8

Which TWO techniques are commonly used for feature selection in machine learning? (Choose 2)

Accepted Answer

L1 regularization (Lasso). L1 regularization (Lasso) is correct because it adds a penalty equal to the absolute value of the magnitude of coefficients, which can shrink some coefficients exactly to zero, effectively performing feature selection by removing irrelevant features from the model. This makes it a built-in feature selection technique within the training process.

Answer

Principal Component Analysis (PCA)

Answer

SMOTE

Answer

Dropout

Question 9

Which THREE are common data preprocessing steps in a machine learning pipeline? (Choose 3)

Accepted Answer

Encoding categorical variables. Encoding categorical variables is a common data preprocessing step because machine learning algorithms require numerical input. Techniques like one-hot encoding or label encoding convert categorical data (e.g., colors, countries) into numeric format, enabling the model to process them correctly. Without this step, the model would misinterpret categorical labels as ordinal or meaningless numeric values.

Answer

Hyperparameter tuning

Answer

Model evaluation

Question 10

Which TWO are best practices for versioning machine learning models? (Choose 2)

Accepted Answer

Tag each model with training date, hyperparameters, and performance metrics. Option B is correct because tagging each model with training date, hyperparameters, and performance metrics creates a reproducible audit trail. This practice aligns with MLOps principles, enabling teams to trace model behavior back to specific training runs and compare versions objectively.

Answer

Use the same model version for all deployments

Answer

Store only the final model binary without metadata

Answer

Manually rename model files with version numbers

Question 11

An engineer is training a neural network and observes the output shown. Which conclusion is most likely correct?

Accepted Answer

The model is overfitting after epoch 2.. The output shows training loss decreasing while validation loss increases after epoch 2, which is a classic sign of overfitting. The model begins to memorize the training data rather than generalize, leading to poor performance on unseen data. This pattern confirms that overfitting starts after epoch 2, making option B correct.

Answer

The gradients are vanishing.

Answer

The model is underfitting.

Answer

The learning rate is too high.

Question 12

A data engineer is reviewing an S3 bucket policy for a machine learning project. The policy is intended to allow access to training data only from the corporate network (10.0.0.0/16). However, users in the corporate network report access denied. Which issue is most likely causing the problem?

Accepted Answer

The policy does not include s3:ListBucket action, which is needed to list objects.. Option C is correct. The condition key uses "aws:SourceIp" but the correct key for IP address condition is "aws:SourceIp". However, the exhibit shows "aws:SourceIp" which is correct? Wait, the exhibit shows "aws:SourceIp" — that is correct. But the problem might be that the policy uses "GetObject" but the users might be trying to list objects (s3:ListBucket). Option C says the action does not include s3:ListBucket, which is often needed first. Option A is wrong because the resource is correct. Option B is wrong because the policy allows access from 10.0.0.0/16. Option D is wrong because the effect is Allow.

Answer

The policy is missing a Deny statement.

Answer

The resource ARN is incorrect.

Answer

The IP address condition is not matching because the corporate network uses a different CIDR.

Question 13

A data scientist is training a deep learning model for image classification. The training loss decreases steadily but the validation loss starts increasing after 10 epochs. Which technique should the scientist apply to address this issue?

Accepted Answer

Implement early stopping. The scenario describes overfitting: the model memorizes training data (loss decreases) but fails to generalize to unseen validation data (validation loss increases). Early stopping (Option C) halts training when validation performance degrades, preventing overfitting while preserving the best model weights. This is a standard regularization technique in deep learning frameworks like TensorFlow and PyTorch.

Answer

Add more dropout layers

Answer

Reduce the learning rate

Answer

Increase the number of training epochs

Question 14

A financial institution is building a fraud detection system using a supervised learning model. The dataset is highly imbalanced with 99.9% legitimate transactions and 0.1% fraudulent ones. Which approach would be MOST effective to train the model to detect fraud?

Accepted Answer

Use SMOTE to generate synthetic fraudulent transactions. SMOTE (Synthetic Minority Oversampling Technique) is the most effective approach because it generates synthetic fraudulent transactions by interpolating between existing minority class samples, thereby balancing the dataset without losing information. This allows the model to learn decision boundaries for fraud detection more effectively than simple undersampling or metric adjustments, especially given the extreme 99.9% vs 0.1% imbalance.

Answer

Train the model using accuracy as the performance metric

Answer

Undersample the legitimate transactions to match the number of fraudulent ones

Answer

Increase the regularization strength in the model

Question 15

A company wants to deploy an AI model for real-time inference on edge devices with limited computational resources. Which model architecture would be MOST suitable?

Accepted Answer

MobileNet. MobileNet is specifically designed for mobile and edge devices using depthwise separable convolutions, which drastically reduce the number of parameters and computational cost while maintaining acceptable accuracy. This makes it the most suitable choice for real-time inference on resource-constrained edge hardware.

Answer

YOLOv4

Answer

ResNet-152

Answer

BERT