I would first analyze the query execution plan to identify the bottlenecks and determine which fields are frequently queried but lacking indexes. Then, I would add appropriate indexes, focusing on composite indexes for multi-column queries and ensuring that the indexes align with the query patterns, particularly considering the read-heavy nature of machine learning applications.
In depth
Optimizing database queries involves understanding how the database engine processes those queries. By examining the execution plan, we can see which operations are taking the most time, like full table scans or key lookups. In machine learning scenarios, where datasets can be large and performance critical, the right indexing can significantly enhance response times. Composite indexes should be created for queries involving multiple columns, while also considering the selectivity of the columns; unique or highly selective columns make better candidates for indexing.
We must also be cautious about over-indexing, as too many indexes can degrade the performance of write operations—an essential consideration in an evolving machine learning model where retraining might require frequent updates to the database. Additionally, database indexing strategies should adapt over time as application usage patterns evolve, necessitating regular review and adjustments to the indexing strategy for optimal performance.
Example scenario
In a recent project, we had a machine learning application that relied on quick predictions from a large user dataset. Initial performance testing revealed that a key query used for fetching user features was taking over three seconds to execute. After analyzing the execution plan, we discovered that the query was scanning the entire table due to missing indexes on the user_id and feature_type columns. By adding a composite index on these two columns, we reduced the query execution time to under 100 milliseconds, significantly improving the user experience and allowing for timely predictions.
Common mistakes
A common mistake is failing to analyze the query execution plan before adding indexes; developers often add indexes based on assumptions without understanding the actual query performance characteristics. This can lead to unnecessary indexes that do not improve performance and instead slow down write operations. Another mistake is overlooking index maintenance; as data changes, fragmentation can occur, and not monitoring or rebuilding indexes can lead to degraded performance over time. It's crucial to approach indexing with a balanced strategy that considers both read and write workloads.
Follow-up questions
- What tools do you use to analyze query performance?
- Can you explain how to determine the optimal number of indexes?
- How would you handle the trade-offs between read and write operations in an indexing strategy?
- Have you ever had to remove an index, and what prompted that decision?

Preparing for a system design interview?
- A 7-step method for answering any system design question, plus estimation numbers
- 5 worked designs: URL shortener, rate limiter, chat, news feed and file storage
- Guides to caching, databases, queues, consistency and reliability, with a 4-week study plan
- 306 practice questions with answers on caching, databases, queues, microservices, APIs and more
5 worked designs, a 4-week plan and 306 practice questions with answers.
The material stays free to read online. This is the organised, offline, print-ready PDF edition. Secure payment by PayPal; your personal download link is emailed after payment is confirmed. Digital product, no shipping.