Mapping resolution has recently been identi ed as a key limitation in successfully locating the drivers of atrial brillation. Using a simple cellular automata model of atrial fibrillation, we demonstrate a method by which re-entrant drivers can be located quickly and accurately using a collection of indirect electrogram measurements. The method proposed employs simple, out of the box machine learning algorithms to correlate characteristic electrogram gradients with the displacement of an electrogram recording from a re-entrant driver. Such a method is less sensitive to local fluctuations in electrical activity. As a result, the method successfully locates 95.4% of drivers in tissues containing a single driver, and 95.1% (92.6%) for the first (second) driver in tissues containing two drivers of atrial fi brillation. Additionally, we demonstrate how the technique can be applied to tissues with an arbitrary number of drivers. In its current form, the techniques presented are not refi ned enough for a clinical setting. However, the methods proposed offer a promising path for future investigations aimed at improving targetted ablation for atrial fibrillation.