Search papers, labs, and topics across Lattice.
MAM-AI is an on-device medical retrieval-augmented generation system designed for nurse-midwives in Zanzibar, addressing the challenge of accessing authoritative guidelines in a resource-limited setting. The system utilizes a 300M parameter embedding model to match questions against a curated corpus of 87 guideline documents and generates answers with a 4B parameter generator, all while operating offline. Evaluation reveals that while the retrieval component performs comparably to cloud systems, the generator's ability to provide safe and helpful responses is limited, necessitating careful selection and prompting strategies to enhance its reliability and fidelity to sources.
On-device retrieval is nearly solved, but ensuring safe and helpful responses from generative models remains a critical challenge in medical AI applications.
Maternal and newborn mortality remain among the highest in sub-Saharan Africa, where midwifery care is often delivered by nurses who lack midwifery training to international standards, and consulting authoritative guidance at the point of care is hard: the guidelines are long and connectivity is intermittent. We present MAM-AI, a medical question-answering assistant for nurse-midwives in Zanzibar that runs entirely on a commodity Android device: a question is embedded (EmbeddingGemma, 300M) and matched against a curated corpus of 87 guideline documents (63,650 passages), then answered with citations by a 4B int4 generator (Gemma 4 E4B), fully offline, with no query leaving the device. We evaluate the exact deployed configuration with a layered methodology -- retriever, generator under oracle context, end-to-end, and latency -- scored by LLM judges validated against physician rubrics. The evaluation relocates the hard problem. On-device retrieval is essentially solved: the 300M embedder ranks third of seven retrievers and rivals cloud systems, so the passages the system needs are usually found. The small generator is what remains in doubt: adding retrieved context does not improve its answers, and at 4B it cannot be both helpful and safe at once -- of two same-size candidates, the more helpful one commits genuine dangerous errors, so we deploy the other, which is about twice as faithful to its sources (as faithful as a frontier model), and recover its helpfulness with a redesigned prompt that cuts deflection from 33% to 3%. Corpus quality is decisive for the same reason: where the corpus holds the right passage the answer is specific and actionable, and where it does not it goes vague. MAM-AI is a thoroughly evaluated, open-source research prototype, not a fielded product; the system, knowledge base, benchmarks, and evaluation harness are released.
