Rajeev Rastogi headshot with map of India
Rajeev Rastogi, vice president of machine learning for Amazon India, and his team work to address the needs of more than 600 million people who are online, who together speak more than 22 languages and 19,500 dialects.
Credit: Glynis Condon

How Rajeev Rastogi’s machine learning team in India develops innovations for customers worldwide

Team works to address the needs of 600 million people online who together speak more than 22 Indian languages with over 19,500 dialects.

As vice president of machine learning at Amazon India, Rajeev Rastogi is helping his team drive innovations that have a profound impact not only on shoppers in India, but also on the company’s customers around the world. For example, models developed by Amazon’s scientists in India have been used globally to improve the quality of Amazon’s catalog by ensuring that for all products, images match with the title. In addition, including delivery speed as a feature in search ranking — a key factor that helps surface ‘faster’ offers to customers in search results — was first launched in Amazon India.

Rastogi began his career at Bell Labs. His early work involved developing clustering algorithms that could scale — a significant innovation in a field that was then dominated by statisticians working on relatively smaller data sets. Rastogi also served as the vice president of Yahoo Labs, where his team developed data-extraction algorithms to pull structured information from billions of webpages, and then present them to users in easily digestible ways.

View from space of a connected network around planet Earth representing the Internet of Things.
Sign up for our newsletter

Rastogi joined Amazon in 2012. His first Amazon project involved the development of algorithms to classify products into Amazon’s large and complex taxonomical structure — for example, to classify a Samsonite luggage set in ‘Carry-On Luggage,’ ‘Suitcases’ and ‘Luggage Sets.’ Since then, Rastogi has been involved in utilizing science to make an impact in a number of areas that have resulted in faster, more seamless and sustainable, shopping experiences.

In this interview, Rastogi spoke about the projects his teams have worked on to improve the shopping experience for Amazon’s customers, a recently developed statistical model that has helped Amazon reduce product-shipment damage in India, and innovations developed to help customers get what they need safely after the outbreak of the COVID-19 pandemic.

Q. What are some of the ways that science has helped improve the shopping experience for Amazon’s customers in India?

India is a unique market in several important ways. There are more than 600 million people online in the country. Many of them are relatively new to digital shopping. Over 85% of our traffic comes from a diverse range of mobile devices.  To complicate matters, mobile customers in India can experience fluctuating speeds due to congested towers and tower switching.

We’ve developed models to predict customers who are on a slow or spotty network based on criteria like device characteristics, cell tower information, and the latency of the last request. For such customers, we provide an adaptive experience and serve streamlined pages with a lower number of widgets that are easier to navigate.

With more than 22 languages and 19,500 dialects, India is also an incredibly diverse country with strong regional preferences. A customer searching for a sari in Gujarat may be interested in a “Bandhani,” which is popular in that state, while a customer in Karnataka searching for a sari may be looking for “Mysore Silk,” a popular variety in that region. To surface regionally popular and relevant products in search results, we have added regional sales for products as a feature in search.

A key problem in India and other emerging countries is that addresses are highly unstructured; they are also incomplete, with critical address fields such as street name missing from the address. For example, we have seen addresses on Amazon.in such as “Near Orion Mall, Malleswaram, Bangalore”, or “Near Bus Stand, Sambhaji Chowk, Nasik”. Our team has developed a machine-learning-based “Address Deliverability Score” to identify poor quality and incomplete addresses that are difficult to locate and deliver to, and intercept them at address creation time to improve address quality.  

You can also have issues related to catalog quality. For example, important attribute values such as the color of a product may be missing for a product. This means that a shoe might be red, and yet might not show up in the list of results for a customer searching for ‘red shoe.’

We use a variety of deep learning models to improve catalog quality by extracting attributes such as color from product titles and images, and backfilling missing product information. To give just one example, we use attention mechanisms to focus the attention of convolutional neural networks on parts of the image from where we want to extract the color of a product. 

We also utilize semi-supervised learning techniques to train neural networks extensively, which greatly reduces the need for large amounts of labeled data. What I love about this approach is that unlabeled data can be a treasure trove of information, particularly for understanding higher-level representations. For example, an algorithm can analyze text patterns around words to understand that ‘car’ and ‘automobile’ are similar without having to explicitly specify that they are synonyms.

India is a market unlike any other in the world, and I’m proud of how we are using science to solve some really difficult problems for our customers.

Q. How are you using science to make Amazon more sustainable?

Amazon has committed to reach net zero carbon by 2040, one decade ahead of the Paris Agreement. Science will play an extremely important role in enabling innovations that will make this happen.

Let me give you just one example. At this year’s European Conference on Machine Learning, members of my team presented a new model for determining the best way to package a given product. We’ve all seen customers not happy about damaged products and excessive product packaging. Incorrect packaging is not only wasteful and bad for the environment, but it also increases our packaging and concessions costs.

India is a market unlike any other in the world, and I’m proud of how we are using science to solve some really difficult problems for our customers.
Rajeev Rastogi

Determining the optimal way to ship a product is complicated. Because one product is rarely shipped across all different package types, you run into situations where there’s a lack of ground truth data. In addition, we have the problem of enforcing ordinality into the process. We have to predict higher probabilities of damage for less expensive (less robust) packaging options, and lower probabilities of damage for more expensive (more robust) options. Enforcing ordinality is not something that standard machine learning techniques do naturally.

The solution developed by my team is as elegant as it is simple. Our scientists developed a linear model, with carefully designed constraints on the model parameters to impose ordinality. To further enforce ordinality, we used data augmentation. This means that for a product-package pair that resulted in product damage, we added examples of that product coupled with less robust packages, also labeled as resulting in damage. 

We’ve applied the model to hundreds of thousands of Amazon packages, reducing shipment damage very significantly while actually saving on shipping costs. This innovation is a testament to the incredible scientific talent at Amazon India. It also speaks volumes of our desire and our ability to take on the really big problems — those that have a significant impact on the lives of our customers and the world at large.

Q. What are some scientific innovations from your team to help customers get what they need safely during COVID-19?

As soon as the pandemic struck, I became interested in what we could do as scientists to keep people safe, and help them get what they need during these trying times. Could we use technology to generate an infection risk score for each individual? These scores could be leveraged by governments and organizations to prioritize testing and identify individuals to quarantine.

We all know that COVID-19 spreads through contacts. Many governments have developed contact tracing apps that use Bluetooth signals on mobile phones to track social contacts among individuals. However, it is challenging to use this fine-grained contact data of individuals to estimate an infection risk score for each individual. This is because the probability of infection transmission through a contact depends on the duration, distance, and location (indoors, outdoors) of the contact. Furthermore, individuals may have indirectly come in contact with a person who has tested positive for COVID-19. Or they may have come in contact with an infected person, but during the period when he or she was not contagious.

I worked with fellow scientists to develop a probabilistic graphical model called CRISP for COVID-19 infection spread through contacts between individuals. The model builds off the SEIR (Susceptible-Exposed-Infectious-Removed) approach that is commonly used to track the different epidemiological status of individuals. Our model captures the transitions between these different states, while also accounting for test outcomes. We developed a block-Gibbs sampling algorithm to draw samples of the latent infection status of each individual, given data about contacts and test results. These infection status samples are then used to compute infection risk scores for each individual. We also developed a Monte Carlo Expectation Maximization (EM) algorithm to infer the infection transmission probability for each contact taking into account factors such as contact duration, distance, and location.  

Also during the pandemic, our operations team built virtual pickup points to deliver packages to customers who live in quarantined apartment buildings. The problem: identifying customers who live in these buildings and educating them about the virtual pickup points. We used address segmentation machine learning models to extract apartment building names from delivery addresses input by customers. We then sent emails to these customers notifying them about the new features. Customers were really excited about this new feature — the email open rates announcing virtual pickup points were higher than 50%.

I’ve been at Amazon for eight years now. I joined Amazon because I was excited at the prospect of conducting scientific work that had the potential to have a real-world impact. And what was true back then remains true today — I come to work every day invigorated at the potential of making a difference in the lives of millions of people around the world.

Research areas

Related content

US, Virtual
The Amazon Economics Team is hiring Interns in Economics. We are looking for detail-oriented, organized, and responsible individuals who are eager to learn how to work with large and complicated data sets. Some knowledge of econometrics, as well as basic familiarity with Stata, R, or Python is necessary. Experience with SQL, UNIX, Sawtooth, and Spark would be a plus. These are full-time positions at 40 hours per week, with compensation being awarded on an hourly basis. You will learn how to build data sets and perform applied econometric analysis at Internet speed collaborating with economists, data scientists and MBAʼs. These skills will translate well into writing applied chapters in your dissertation and provide you with work experience that may help you with placement. Roughly 85% of interns from previous cohorts have converted to full time economics employment at Amazon. If you are interested, please send your CV to our mailing list at econ-internship@amazon.com.
US, WA, Seattle
Amazon internships are full-time (40 hours/week) for 12 consecutive weeks with start dates in May - July 2023. Our internship program provides hands-on learning and building experiences for students who are interested in a career in hardware engineering. This role will be based in Seattle, and candidates must be willing to work in-person. Corporate Projects (CPT) is a team that sits within the broader Corporate Development organization at Amazon. We seek to bring net-new, strategic projects to life by working together with customers and evolving projects from ZERO-to-ONE. To do so, we deploy our resources towards proofs-of-concept (POCs) and pilot programs and develop them from high-level ideas (the ZERO) to tangible short-term results that provide validating signal and a path to scale (the ONE). We work with our customers to develop and create net-new opportunities by relentlessly scouring all of Amazon and finding new and innovative ways to strengthen and/or accelerate the Amazon Flywheel. CPT seeks an Applied Science intern to work with a diverse, cross-functional team to build new, innovative customer experiences. Within CPT, you will apply both traditional and novel scientific approaches to solve and scale problems and solutions. We are a team where science meets application. A successful candidate will be a self-starter comfortable with ambiguity, strong attention to detail, and the ability to work in a fast-paced, ever-changing environment. As an Applied Science Intern, you will own the design and development of end-to-end systems. You’ll have the opportunity to create technical roadmaps, and drive production level projects that will support Amazon Science. You will work closely with Amazon scientists, and other science interns to develop solutions and deploy them into production. The ideal scientist must have the ability to work with diverse groups of people and cross-functional teams to solve complex business problems.
US, WA, Seattle
Amazon internships are full-time (40 hours/week) for 12 consecutive weeks with start dates in May - July 2023. Our internship program provides hands-on learning and building experiences for students who are interested in a career in hardware engineering. This role will be based in Seattle, and candidates must be willing to work in-person. Corporate Projects (CPT) is a team that sits within the broader Corporate Development organization at Amazon. We seek to bring net-new, strategic projects to life by working together with customers and evolving projects from ZERO-to-ONE. To do so, we deploy our resources towards proofs-of-concept (POCs) and pilot programs and develop them from high-level ideas (the ZERO) to tangible short-term results that provide validating signal and a path to scale (the ONE). We work with our customers to develop and create net-new opportunities by relentlessly scouring all of Amazon and finding new and innovative ways to strengthen and/or accelerate the Amazon Flywheel. CPT seeks an Applied Science intern to work with a diverse, cross-functional team to build new, innovative customer experiences. Within CPT, you will apply both traditional and novel scientific approaches to solve and scale problems and solutions. We are a team where science meets application. A successful candidate will be a self-starter comfortable with ambiguity, strong attention to detail, and the ability to work in a fast-paced, ever-changing environment. As an Applied Science Intern, you will own the design and development of end-to-end systems. You’ll have the opportunity to create technical roadmaps, and drive production level projects that will support Amazon Science. You will work closely with Amazon scientists, and other science interns to develop solutions and deploy them into production. The ideal scientist must have the ability to work with diverse groups of people and cross-functional teams to solve complex business problems.
US, MA, Westborough
Are you inspired by invention? Is problem solving through teamwork in your DNA? Do you like the idea of seeing how your work impacts the bigger picture? Answer yes to any of these and you’ll fit right in here at Amazon Robotics. We are a smart team of doers that work passionately to apply cutting edge advances in robotics and software to solve real-world challenges that will transform our customers’ experiences in ways we can’t even imagine yet. We invent new improvements every day. We are Amazon Robotics and we will give you the tools and support you need to invent with us in ways that are rewarding, fulfilling and fun. Amazon Robotics is seeking interns and co-ops with a passion for robotic research to work on cutting edge algorithms for robotics. Our team works on challenging and high-impact projects, including allocating resources to complete a million orders a day, coordinating the motion of thousands of robots, autonomous navigation in warehouses, identifying objects and damage, and learning how to grasp all the products Amazon sells. We are seeking internship candidates with backgrounds in computer vision, machine learning, resource allocation, discrete optimization, search, and planning/scheduling. You will be challenged intellectually and have a good time while you are at it! Key job responsibilities • Identifying creative solutions for challenging research problems in robotics and computer vision • Developing software solutions to test hypotheses and demonstrate new functionality • Prototyping concepts to collect data and measure performance • Writing code and unit tests and integrating code with other software and hardware components • Utilizing Amazon Robotics and Amazon engineering tools, processes and technologies • Delivering a final presentation to managers and engineers on the successes and challenges of their internship and the business value they have contributed
US, CA, Palo Alto
The Amazon Search team creates powerful, customer-focused search solutions and technologies. Whenever a customer visits an Amazon site worldwide and types in a query or browses through product categories, Amazon Search services go to work. We design, develop, and deploy high performance, fault-tolerant distributed search systems used by millions of Amazon customers every day. We’re seeking a Principal Scientist with a deep expertise in Search Science. Your responsibilities will include everything from developing and prototyping innovative machine learning, and deep learning algorithms to implementing, testing, and supporting full solutions in a production environment. We are looking for innovators who can contribute to advancing search technology on what’s scientifically possible while remaining committed to creating world-class products. Joining this team, you’ll experience the benefits of working in a dynamic, entrepreneurial environment, while leveraging the resources of Amazon.com (AMZN), Earth's most customer-centric company one of the world's leading internet companies. We provide a highly customer-centric, team-oriented environment in our offices located in Palo Alto, California. Key job responsibilities As a hands-on leader of this team, you’ll be responsible for defining key research questions, identifying relevant data, adopting or proposing innovative machine learning solutions conducting rigorous experiments, publishing results and working with the engineering team to deploy these solutions. As a strategic leader, you will identify investment opportunities, develop long term strategies, and propose, prioritize and deliver on goals. You’ll also participate in organizational planning, hiring, mentorship and leadership development. You will be technically fearless and with a passion for building scalable science and engineering solutions. You will serve as a key scientific resource in full-cycle development (conception, design, implementation, testing to documentation, delivery, and maintenance). About the team Starting in 2009, the Visual Search & Augmented Reality team has thus far launched many visual search solutions on the Amazon App that use computer vision and machine learning/deep learning to help customers complete their shopping missions more easily; multiple internal teams at Amazon (devices, Kindle, Seller services, etc.) also use our libraries and APIs to deliver solutions to their own customers. We are a full stack shop, and our team capabilities cover the whole solution spectrum, ranging across applied science, large scale engineering services, product management, UX design, and mobile app development for iOS and Android.
US, MN, Minneapolis
AWS Central Economics is an interdisciplinary team on the cutting edge of economics, statistical analysis, and machine learning whose mission is to solve problems that have high risk with abnormally high returns. Our team leverages the strengths of our scientists to build solutions for some of the toughest business problems here at Amazon AWS. We are looking for an exceptionally talented, seasoned, and motivated Economist to manage a team of economists and data scientists to drive the science for AWS. Key job responsibilities Manage a team of economists and data scientists to deliver actionable economic analyses to business leaders, provide leadership on the economics and science used in the analyses, and engage with business leaders to identify challenges AWS faces that call for in-depth economic analyses and to ensure the analyses have their intended impact.
LU, Luxembourg
&ltHire Relocation Requisition - not for posting> Provides insights to leadership on improving Supply Chain cost and Speed by using Data Science and Analytics techniques. Build Dashboards and models to industrialize these findings at scale.
US, VA, Arlington
The People eXperience and Technology Central Science Team (PXTCS) uses economics, behavioral science, statistics, and machine learning to proactively identify mechanisms and process improvements which simultaneously improve Amazon and the lives, wellbeing, and the value of work to Amazonians. We are an interdisciplinary team that combines the talents of science and engineering to develop and deliver solutions that measurably achieve this goal. We are looking for economists who are able to work with business partners to hone complex problems into specific, scientific questions, and test those questions to generate insights. The ideal candidate will work with engineers and computer scientists to estimate models and algorithms on large scale data, design pilots and measure their impact, and transform successful prototypes into improved policies and programs at scale. We are looking for creative thinkers who can combine a strong technical economic toolbox with a desire to learn from other disciplines, and who know how to execute and deliver on big ideas as part of an interdisciplinary technical team. Ideal candidates will work closely with business partners to develop science that solves the most important business challenges. They will work in a team setting with individuals from diverse disciplines and backgrounds. They will serve as an ambassador for science and a scientific resource for business teams, so that scientific processes permeate throughout the HR organization to the benefit of Amazonians and Amazon. Ideal candidates will own the data analysis, modeling, and experimentation that is necessary for estimating and validating models. They will work closely with engineering teams to develop scalable data resources to support rapid insights, and take successful models and findings into production as new products and services. They will be customer-centric and will communicate scientific approaches and findings to business leaders, listening to and incorporate their feedback, and delivering successful scientific solutions. Key job responsibilities Use causal inference methods to evaluate the impact of policies on employee outcomes. Examine how external labor market and economic conditions impact Amazon's ability to hire and retain talent. Use scientifically rigorous methods to develop and recommend career paths for employees. A day in the life Work with teammates to apply economic methods to business problems. This might include identifying the appropriate research questions, writing code to implement a DID analysis or estimate a structural model, or writing and presenting a document with findings to business leaders. Our economists also collaborate with partner teams throughout the process, from understanding their challenges, to developing a research agenda that will address those challenges, to help them implement solutions. About the team We are a multidisciplinary team that combines the talents of science and engineering to develop innovative solutions to make Amazon Earth's Best Employer.
US, WA, Seattle
The People eXperience and Technology Central Science Team (PXTCS) uses economics, behavioral science, statistics, and machine learning to proactively identify mechanisms and process improvements which simultaneously improve Amazon and the lives, wellbeing, and the value of work to Amazonians. We are an interdisciplinary team that combines the talents of science and engineering to develop and deliver solutions that measurably achieve this goal. We are looking for economists who are able to apply economic methods to address business problems. The ideal candidate will work with engineers and computer scientists to estimate models and algorithms on large scale data, design pilots and measure their impact, and transform successful prototypes into improved policies and programs at scale. We are looking for creative thinkers who can combine a strong technical economic toolbox with a desire to learn from other disciplines, and who know how to execute and deliver on big ideas as part of an interdisciplinary technical team. Ideal candidates will work in a team setting with individuals from diverse disciplines and backgrounds. They will work with teammates to develop scientific models and conduct the data analysis, modeling, and experimentation that is necessary for estimating and validating models. They will work closely with engineering teams to develop scalable data resources to support rapid insights, and take successful models and findings into production as new products and services. They will be customer-centric and will communicate scientific approaches and findings to business leaders, listening to and incorporate their feedback, and delivering successful scientific solutions. Key job responsibilities Use causal inference methods to evaluate the impact of policies on employee outcomes. Examine how external labor market and economic conditions impact Amazon's ability to hire and retain talent. Use scientifically rigorous methods to develop and recommend career paths for employees. A day in the life Work with teammates to apply economic methods to business problems. This might include identifying the appropriate research questions, writing code to implement a DID analysis or estimate a structural model, or writing and presenting a document with findings to business leaders. Our economists also collaborate with partner teams throughout the process, from understanding their challenges, to developing a research agenda that will address those challenges, to help them implement solutions. About the team We are a multidisciplinary team that combines the talents of science and engineering to develop innovative solutions to make Amazon Earth's Best Employer.
US, WA, Seattle
Amazon is looking for talented Postdoctoral Scientists to join our global Science teams for a one-year, full-time research position. Postdoctoral Scientists will innovate as members of Amazon’s key global Science teams, including: AWS, Alexa AI, Alexa Shopping, Amazon Style, CoreAI, Last Mile, and Supply Chain Optimization Technologies. Postdoctoral Scientists will join one of may central, global science teams focused on solving research-intense business problems by leveraging Machine Learning, Econometrics, Statistics, and Data Science. Postdoctoral Scientists will work at the intersection of ML and systems to solve practical data driven optimization problems at Amazon scale. Postdocs will raise the scientific bar across Amazon by diving deep into exploratory areas of research to enhance the customer experience and improve efficiencies. Please note: This posting is one of several Amazon Postdoctoral Scientist postings. Please only apply to a maximum of 2 Amazon Postdoctoral Scientist postings that are relevant to your technical field and subject matter expertise. Key job responsibilities * Work closely with a senior science advisor, collaborate with other scientists and engineers, and be part of Amazon’s vibrant and diverse global science community. * Publish your innovation in top-tier academic venues and hone your presentation skills. * Be inspired by challenges and opportunities to invent cutting-edge techniques in your area(s) of expertise.