AI/machine learning

“Doing Machine Learning the Right Way”

MIT Professor Aleksander Madry strives to build machine-learning models that are more reliable, understandable, and robust.

madry.jpg
Alexander Madry.
Credit: Ian MacLellan/MIT.

The work of Massachusetts Institute of Technology (MIT) computer scientist Aleksander Madry is fueled by one core mission: “doing machine learning the right way.”

Madry’s research centers largely on making machine learning—a type of artificial intelligence—more accurate, efficient, and robust against errors. In his classroom and beyond, he also worries about questions of ethical computing as we approach an age where artificial intelligence will have great impact on many sectors of society.

"I want society to truly embrace machine learning," said Madry, a recently tenured professor in the Department of Electrical Engineering and Computer Science. "To do that, we need to figure out how to train models that people can use safely, reliably, and in a way that they understand."

Interestingly, his work with machine learning dates back only a couple of years to shortly after he joined MIT in 2015. In that time, his research group has published several critical papers demonstrating that certain models can be easily tricked to produce inaccurate results—and showing how to make them more robust.

In the end, he aims to make each model’s decisions more interpretable by humans so researchers can peer inside to see where things went awry. At the same time, he wants to enable nonexperts to deploy the improved models in the real world for, say, helping diagnose disease or control driverless cars.

"It’s not just about trying to crack open the machine-learning black box. I want to open it up, see how it works, and pack it back up so people can use it without needing to understand what’s going on inside," he said.

For the Love of Algorithms

Madry was born in Wroclaw, Poland, where he attended the University of Wroclaw as an undergraduate in the mid-2000s. While he harbored interest in computer science and physics, "I actually never thought I’d become a scientist," he said.

An avid video gamer, Madry initially enrolled in the computer science program with intentions of programming his own games. But, in joining friends in a few classes in theoretical computer science and, in particular, theory of algorithms, he fell in love with the material. Algorithm theory aims to find efficient optimization procedures for solving computational problems, which requires tackling difficult mathematical questions. “I realized I enjoy thinking deeply about something and trying to figure it out,” said Madry, who wound up double-majoring in physics and computer science.

When it came to delving deeper into algorithms in graduate school, he went to his first choice: MIT. There, he worked under both Michel X. Goemans, who was a major figure in applied math and algorithm optimization, and Jonathan A. Kelner, who had just arrived to MIT as a junior faculty working in that field. For his PhD dissertation, Madry developed algorithms that solved a number of longstanding problems in graph algorithms, earning the 2011 George M. Sprowls Doctoral Dissertation Award for the best MIT doctoral thesis in computer science.

After earning his PhD, Madry spent a year as a postdoc at Microsoft Research New England before teaching for 3 years at the Swiss Federal Institute of Technology Lausanne—which Madry calls "the Swiss version of MIT." But his alma mater kept calling him back. "MIT has the thrilling energy I was missing," he said. "It’s in my DNA."

Getting Adversarial

Shortly after joining MIT, Madry found himself swept up in a novel science: machine learning. In particular, he focused on understanding the re-emerging paradigm of deep learning. That’s an artificial-intelligence application that uses multiple computing layers to extract high-level features from raw input—such as using pixel-level data to classify images. MIT’s campus was, at the time, buzzing with new innovations in the domain.

But that led to the question: Was machine learning all hype or solid science? "It seemed to work, but no one actually understood how and why," Madry said.

Answering that question set his group on a long journey, running experiment after experiment on deep-learning models to understand the underlying principles. A major milestone in this journey was an influential paper they published in 2018 that presented a methodology for making machine-learning models more resistant to "adversarial examples." Adversarial examples are slight perturbations to input data that are imperceptible to humans—such as changing the color of one pixel in an image—but that cause a model to make inaccurate predictions. They illuminate a major shortcoming of existing machine-learning tools.

Continuing this line of work, Madry’s group showed that the existence of these mysterious adversarial examples may contribute to how machine-learning models make decisions. In particular, models designed to differentiate images of, say, cats and dogs, make decisions based on features that do not align with how humans make classifications. Simply changing these features can make the model consistently misclassify cats as dogs without changing anything in the image that’s really meaningful to humans.

Results indicated some models—which may be used, for example, to identify abnormalities in medical images or help autonomous cars identify objects in the road—aren’t exactly up to snuff. "People often think these models are superhuman, but they didn’t actually solve the classification problem we intend them to solve," Madry said. "And their complete vulnerability to adversarial examples was a manifestation of that fact. That was an eye-opening finding."

Read the full story here.